Perpustakaan Pelatihan Model AutoML & PyTorch sumber terbuka
Dokumen | Contoh
!!! perhatian GradsFlow berubah dengan cepat. Akan ada banyak perubahan besar hingga kita mencapai 0.1.0
. Jangan ragu untuk memberikan masukan Anda dengan membuat masalah atau bergabunglah dengan grup Slack kami.
GradsFlow adalah Perpustakaan AutoML sumber terbuka berdasarkan PyTorch. Tujuan kami adalah mendemokratisasi AI dan membuatnya tersedia bagi semua orang.
Itu dapat secara otomatis membangun & melatih Model Pembelajaran Mendalam untuk berbagai tugas di laptop Anda atau ke cluster jarak jauh langsung dari laptop Anda. Ini menyediakan API Pelatihan Model yang kuat dan mudah diperluas yang dapat digunakan untuk melatih hampir semua model PyTorch. Meskipun GradsFlow memiliki API Pelatihan Modelnya sendiri, GradsFlow juga mendukung PyTorch Lightning Flash untuk menyediakan fitur yang lebih kaya di berbagai tugas.
!!! info Gradsflow dibuat untuk pemula dan ahli ! AutoTasks
menyediakan AutoML tanpa kode sementara Model
dan Tuner
menyediakan pelatihan model khusus dan pengoptimalan Hyperparameter.
Direkomendasikan :
Metode yang disarankan untuk menginstal gradsflow
adalah dengan pip
dari PyPI atau, dengan conda
dari saluran conda-forge.
dengan pip
pip install -U gradsflow
dengan konda
conda install -c conda-forge gradsflow
Terbaru (tidak stabil):
Anda juga dapat menginstal gradsflow
versi terbaru (mungkin tidak stabil), jika Anda merasa cukup termotivasi, sebagai berikut:
pip install git+https://github.com/gradsflow/gradsflow@main
Apakah Anda seorang pemula atau berlatar belakang non Machine Learning? Bagian ini untuk Anda. Gradsflow AutoTask
menyediakan pembuatan dan pelatihan model otomatis untuk berbagai tugas berbeda termasuk Pengenalan Gambar, Analisis Sentimen, Peringkasan Teks, dan banyak lagi yang akan datang.
Tuner
menyediakan API yang disederhanakan untuk berpindah dari Pelatihan Model ke pengoptimalan Hyperparameter.
gradsflow.core
: Inti mendefinisikan blok penyusun tugas AutoML.
gradsflow.autotasks
: AutoTasks mendefinisikan berbagai tugas ML/DL yang disediakan oleh gradsflow AutoML API.
gradsflow.model
: GradsFlow Model menyediakan API Pelatihan Model yang sederhana namun dapat disesuaikan. Anda dapat melatih model PyTorch apa pun menggunakan model.fit(...)
dan model ini mudah disesuaikan untuk tugas yang lebih kompleks.
gradsflow.tuner
: Pencarian AutoModel HyperParameter dengan sedikit perubahan kode.
? Lihat contoh buku catatan untuk mempelajari lebih lanjut.
? Mensponsori ko-fi
? Apakah Anda memerlukan dukungan? Hubungi kami di [email protected]
Sosial : Anda juga dapat mengikuti kami di Twitter @gradsflow dan Linkedin untuk pembaruan terkini.
Bergabunglah dengan grup Slack untuk mengobrol dengan kami.
Kontribusi dalam bentuk apa pun dipersilakan. Anda dapat memperbarui dokumentasi, menambahkan contoh, memperbaiki masalah yang teridentifikasi, menambah/meminta fitur baru.
Untuk rincian lebih lanjut, lihat Pedoman Berkontribusi sebelum berkontribusi.
Kami berjanji untuk bertindak dan berinteraksi dengan cara yang berkontribusi terhadap komunitas yang terbuka, ramah, beragam, inklusif, dan sehat.
Baca Kode Etik Perjanjian Kontributor selengkapnya
GradsFlow dibangun dengan bantuan proyek sumber terbuka yang mengagumkan (termasuk namun tidak terbatas pada) Ray, PyTorch Lightning, HuggingFace Accelerate, TorchMetrics. Dibutuhkan inspirasi dari berbagai proyek Keras & FastAI.