機器之心報道
編輯:陳萍
經過數月的封閉測試後,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 及其生态系統的研究和訓練,隻需單擊幾下即可直覺地完成大多數任務。
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 等相容。
TorchStudio 還能進行可視化,無論使用者輸入 / 輸出的數量和張量格式如何,都可以按需顯示資料集樣本:
TorchStudio 還能生成有關整個資料集的統計報告:
對模型進行可視化,以揭示内部節點如何連接配接以及節點之間的張量大小如何變化:
使用者還能設定網絡參數和超參數,實時監控損失和度量曲線并輸出預測:
此外,TorchStudio 還能對訓練模型進行比較,幫助使用者快速确定最佳模型:
使用 TorchStudio,使用者還能自行編輯或添加子產品,每個元件都是可定制的,包括:資料集、模型、分析器、渲染器、損失、度量、優化器等:
面對如此優秀的編譯器,網友也不禁感歎:「非常漂亮的作品!除了 PyCharm 之外,我不會使用其他 IDE,但我會試一試這個 TorchStudio!」
使用 PyTorch 做研究的小夥伴,可以上手試試這個功能齊全的編譯器。