Библиотека обучения моделей AutoML и PyTorch с открытым исходным кодом
Документы | Примеры
!!! внимание GradsFlow быстро меняется. Пока мы не достигнем 0.1.0
, будет много критических изменений. Не стесняйтесь высказать свое мнение, создав задачу или присоединившись к нашей группе Slack.
GradsFlow — это библиотека AutoML с открытым исходным кодом, основанная на PyTorch. Наша цель — демократизировать ИИ и сделать его доступным для всех.
Он может автоматически создавать и обучать модели глубокого обучения для различных задач на вашем ноутбуке или в удаленном кластере непосредственно с вашего ноутбука. Он предоставляет мощный и простой в расширении API обучения модели, который можно использовать для обучения практически любой модели PyTorch. Хотя GradsFlow имеет собственный API для обучения моделей, он также поддерживает PyTorch Lightning Flash, предоставляя более широкие возможности для различных задач.
!!! информация Gradsflow создан как для новичков , так и для экспертов ! AutoTasks
обеспечивает AutoML с нулевым кодом, а Model
and Tuner
обеспечивает обучение пользовательской модели и оптимизацию гиперпараметров.
Рекомендуется :
Рекомендуемый метод установки gradsflow
— либо с помощью pip
из PyPI, либо с помощью conda
из канала conda-forge.
с пипсом
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
: Core определяет строительные блоки задач AutoML.
gradsflow.autotasks
: AutoTasks определяет различные задачи ML/DL, которые предоставляются API gradsflow AutoML.
gradsflow.model
: Модель GradsFlow предоставляет простой, но настраиваемый API обучения модели. Вы можете обучить любую модель PyTorch, используя model.fit(...)
и ее легко настроить для более сложных задач.
gradsflow.tuner
: поиск гиперпараметров AutoModel с минимальными изменениями кода.
? Чтобы узнать больше, ознакомьтесь с примерами блокнотов.
? Спонсор на ко-фи
? Вам нужна поддержка? Свяжитесь с нами по адресу [email protected]
Социальные сети : вы также можете следить за нами в Twitter @gradsflow и Linkedin, чтобы быть в курсе последних обновлений.
Присоединяйтесь к группе Slack, чтобы общаться с нами.
Приветствуются любые вклады. Вы можете обновить документацию, добавить примеры, исправить выявленные проблемы, добавить/запросить новую функцию.
Для получения более подробной информации ознакомьтесь с Руководством по участию, прежде чем вносить вклад.
Мы обязуемся действовать и взаимодействовать таким образом, чтобы способствовать созданию открытого, гостеприимного, разнообразного, инклюзивного и здорового сообщества.
Прочтите полный текст Кодекса поведения участников Соглашения.
GradsFlow создан с помощью замечательных проектов с открытым исходным кодом (включая, помимо прочего) Ray, PyTorch Lightning, HuggingFace Accelerate, TorchMetrics. Он черпает вдохновение из нескольких проектов Keras и FastAI.