開源 AutoML 和 PyTorch 模型訓練庫
文檔 |範例
!!!注意 GrassFlow 正在快速變化。在我們達到0.1.0
之前,將會有許多重大變化。請隨時透過建立問題或加入我們的 Slack 小組來提供回饋。
GradsFlow 是一個基於 PyTorch 的開源 AutoML 函式庫。我們的目標是使人工智慧民主化並讓每個人都能使用它。
它可以自動為您的筆記型電腦上的不同任務建立和訓練深度學習模型,或直接從您的筆記型電腦建立和訓練遠端叢集。它提供了強大且易於擴展的模型訓練 API,可用於訓練幾乎任何 PyTorch 模型。儘管 GradsFlow 有自己的模型訓練 API,但它也支援 PyTorch Lightning Flash,以在不同的任務中提供更豐富的功能。
!!! info Gradsflow 是為初學者和專家打造的! AutoTasks
提供零程式碼 AutoML,而Model
和Tuner
提供自訂模型訓練和超參數最佳化。
受到推崇的:
安裝gradsflow
的建議方法是使用 PyPI 中的pip
或使用 conda-forge 通道中的conda
。
有點
pip install -U gradsflow
與康達
conda install -c conda-forge gradsflow
最新(不穩定):
如果您有足夠的動力,您還可以安裝gradsflow
的最新前沿版本(可能不穩定),如下所示:
pip install git+https://github.com/gradsflow/gradsflow@main
您是初學者還是非機器學習背景?本節適合您。 Gradsflow AutoTask
提供跨各種不同任務的自動模型建立和訓練,包括影像辨識、情緒分析、文字摘要等。
Tuner
提供了一個簡化的 API,用於從模型訓練轉向超參數最佳化。
gradsflow.core
:核心定義了 AutoML 任務的建構塊。
gradsflow.autotasks
:AutoTasks 定義了由 gradsflow AutoML API 提供的不同 ML/DL 任務。
gradsflow.model
:GradsFlow 模型提供了一個簡單但可自訂的模型訓練 API。您可以使用model.fit(...)
訓練任何 PyTorch 模型,並且可以輕鬆自訂更複雜的任務。
gradsflow.tuner
:使用最少的程式碼變更進行 AutoModel 超參數搜尋。
?查看筆記本範例以了解更多資訊。
? ko-fi 贊助商
?您需要支援嗎?聯絡我們:[email protected]
社交:您也可以在 Twitter @gradsflow 和 Linkedin 上關注我們,以了解最新動態。
加入 Slack 群組與我們聊天。
歡迎任何形式的貢獻。您可以更新文件、新增範例、修復已識別的問題、新增/要求新功能。
有關更多詳細信息,請在貢獻之前查看貢獻指南。
我們承諾以有助於建立開放、熱情、多元化、包容和健康社區的方式行事和互動。
閱讀完整的貢獻者契約行為準則
GradsFlow是在出色的開源專案(包括但不限於)Ray、PyTorch Lightning、HuggingFace Accelerate、TorchMetrics 的幫助下建立的。它從 Keras 和 FastAI 的多個項目中汲取靈感。