天天看點

PyTorch專用編譯器:跨三大作業系統,單擊幾下完成任務,還帶教程

機器之心報道

編輯:陳萍

經過數月的封閉測試後,TorchStudio 現在終于迎來了公開測試階段。

編譯器的出現,可謂給廣大研究者帶來了極大的便利。簡單講,編譯器就是将一種語言(通常為進階語言)翻譯為另一種語言(通常為低級語言)的程式。一個現代編譯器的主要工作流程包括:源代碼 (source code) 預處理器 (preprocessor) 編譯器 (compiler) 目标代碼 (object code) 連結器 (Linker) 可執行程式 (executables)。

我們經常聽到或用到的編譯器有 Visual Studio、Dev C++、Visual C++ 6.0、GCC、Clang 等。毫不誇張的說正是編譯器的存在,才讓人類語言轉換為計算機可以了解的 0 和 1。去年編譯器神書「龍書」作者 Alfred Vaino Aho 和 Jeffrey David Ullman 還獲得了 2020 圖靈獎。

今日,開源軟體愛好者、音頻編輯軟體 Steinberg SpectraLayers 的建立者 Robin Lobel 釋出了一個免費的開源 PyTorch IDE,名為 TorchStudio,現在是公開測試版 0.9.1 版本。該 IDE 能夠極大地簡化 PyTorch 及其生态系統的研究和訓練,隻需單擊幾下即可直覺地完成大多數任務。

PyTorch專用編譯器:跨三大作業系統,單擊幾下完成任務,還帶教程

TorchStudio 現在支援 Windows 10 或更高版本、macOS 10.14 或更高版本、Ubuntu 20.04 或更高版本。一開始不會使用的小夥伴也不用擔心,TorchStudio 提供了示例教程,新手可以根據教程進行學習,值得一提的是,TorchStudio 官網還貼心的提供了視訊教程,使用者不必擔心自己不會用。

TorchStudio 位址:https://torchstudio.ai/getstarted/

這個 IDE 有啥特點呢?根據官網顯示,它的特征可總結為三個:連結數千個資料集、上千種模型,進行可視化,靈活可擴充。

資料集:浏覽和加載來自 TorchVision、TorchAudio 或 Hugging Face Hub 的數千個資料集(即将推出),包括 CIFAR10、CIFAR100、Caltech101、CelebA 等。

模型:浏覽和訓練來自 TorchVision、TorchAudio、PyTorch Hub (即将推出)或 Hugging Face Hub (即将推出)的數千個模型,包括大家熟悉的 AlexNet、DenseNet、GoogLeNet 等。

使用者還可以在本地或遠端、本地或雲端存儲資料集和訓練模型。此外 TorchStudio 還與 AWS、Azure、Google Cloud 等相容。

PyTorch專用編譯器:跨三大作業系統,單擊幾下完成任務,還帶教程

TorchStudio 還能進行可視化,無論使用者輸入 / 輸出的數量和張量格式如何,都可以按需顯示資料集樣本:

PyTorch專用編譯器:跨三大作業系統,單擊幾下完成任務,還帶教程

TorchStudio 還能生成有關整個資料集的統計報告:

PyTorch專用編譯器:跨三大作業系統,單擊幾下完成任務,還帶教程

對模型進行可視化,以揭示内部節點如何連接配接以及節點之間的張量大小如何變化:

PyTorch專用編譯器:跨三大作業系統,單擊幾下完成任務,還帶教程

使用者還能設定網絡參數和超參數,實時監控損失和度量曲線并輸出預測:

PyTorch專用編譯器:跨三大作業系統,單擊幾下完成任務,還帶教程

此外,TorchStudio 還能對訓練模型進行比較,幫助使用者快速确定最佳模型:

PyTorch專用編譯器:跨三大作業系統,單擊幾下完成任務,還帶教程

使用 TorchStudio,使用者還能自行編輯或添加子產品,每個元件都是可定制的,包括:資料集、模型、分析器、渲染器、損失、度量、優化器等:

PyTorch專用編譯器:跨三大作業系統,單擊幾下完成任務,還帶教程

面對如此優秀的編譯器,網友也不禁感歎:「非常漂亮的作品!除了 PyCharm 之外,我不會使用其他 IDE,但我會試一試這個 TorchStudio!」

PyTorch專用編譯器:跨三大作業系統,單擊幾下完成任務,還帶教程

使用 PyTorch 做研究的小夥伴,可以上手試試這個功能齊全的編譯器。

繼續閱讀