PyTorch Hub發(fā)布獲Yann LeCun強(qiáng)推 一行代碼調(diào)用經(jīng)典模型

近日，F(xiàn)acebook PyTorch 團(tuán)隊(duì)推出了全新 API PyTorch Hub，提供模型的基本構(gòu)建模塊，用于提高機(jī)器學(xué)習(xí)研究的模型復(fù)現(xiàn)性。PyTorch Hub 包含一個經(jīng)過預(yù)訓(xùn)練的模型庫，內(nèi)置對Colab的支持，而且能夠與Papers With Code集成。另外重要的一點(diǎn)是，它的整個工作流程大大簡化。

簡化到什么程度呢？Facebook 首席 AI 科學(xué)家Yann LeCun 兼圖靈獎圖靈獎得主Yann LeCun發(fā)表 Twitter強(qiáng)烈推薦，使用PyTorch Hub，無論是ResNet、BERT、GPT、VGG、PGAN 還是 MobileNet 等經(jīng)典模型，只需輸入一行代碼，就能實(shí)現(xiàn)一鍵調(diào)用。

Twitter 一發(fā)，立刻引來眾多網(wǎng)友評論點(diǎn)贊，并有網(wǎng)友表示希望看到PyTorch Hub 與TensorFlow Hub的區(qū)別。

這個模型聚合中心到底如何呢？我們來一探究竟。

模型復(fù)現(xiàn)是許多領(lǐng)域的基本要求，尤其是在與機(jī)器學(xué)習(xí)相關(guān)的鄰域中。然而，許多機(jī)器學(xué)習(xí)相關(guān)的出版物，要么不可復(fù)現(xiàn)，要么難以復(fù)現(xiàn)。隨著出版物數(shù)量的不斷增長（包括在 arXiv 上發(fā)表的成數(shù)萬篇論文，以及會議提交的大量論文），模型復(fù)現(xiàn)比以往任何時(shí)候都更加重要。雖然這些出版物大多數(shù)都包含代碼和訓(xùn)練好的模型，但如果用戶想復(fù)現(xiàn)這些模型，還需要做大量的額外的工作。

今天，我們很榮幸地宣布推出 PyTorch Hub，它是一個非常簡單的API，并且具有極其簡單的工作流程。它提供模型的基本構(gòu)建模塊，用于提高機(jī)器學(xué)習(xí)研究的模型復(fù)現(xiàn)性。PyTorch Hub 包含一個經(jīng)過預(yù)訓(xùn)練的模型庫，專門用于促進(jìn)研究的可重復(fù)性和快速開展新的研究。PyTorch Hub 內(nèi)置了對 Colab的 支持，并且能夠與 Papers With Code 集成。目前 PyTorch Hub 已包含一系列廣泛的模型，包括分類器和分割器、生成器、變換器等。