⚡ PyTorch Lightning — революция или просто удобный синтаксический сахар?
Этот репозиторий поможет вам разобраться на практике!
Advanced Training Lab — это проект, созданный для демонстрации возможностей PyTorch Lightning ⚡ и ClearML 🛠️ при обучении моделей машинного обучения и глубокого обучения.
📁 src/ — основной код проекта (модель, датасет, тренировка, тестирование).
📁 src/data/ — скрипты для загрузки и инициализации дата модулей.
📁 src/models/ — скрипты для создания моделей.
📁 src/test/ — скрипты для тестирования моделей.
📁 src/utils/ — вспомогательные функции.
🐍 src/train_sign_language.py — скрипт для обучения модели на датасете жестового языка.
🐍 src/demo_sign_language.py — скрипт для инференса модели на одном примере датасета жестового языка.
📄 requirements.txt — список зависимостей.
📄 README.md — это файл, который Вы сейчас читаете.
1️⃣ Клонируем проект из удаленного репозитория 🛠️:
git clone https://github.com/Vlad15lav/advanced-training-lab.git
cd ./advanced-training-lab2️⃣ Установите зависимости 📦:
pip install -r requirements.txt3️⃣ Перейдите в директорию с кодом 🗂️:
cd ./src4️⃣ Запустите обучение модели 🎓:
4️⃣.1️⃣ Запуск обучение модели перевода с языка жестов на английский.
Параметр fast-dev-run для проверки корректности пайплайна обучения.
python train_sign_language.py --fast-dev-run4️⃣.2️⃣ Запуск обучение модели генерации рукописных цифр.
python train_digits_generator.py --epoch 20 --debug-samples-epoch 25️⃣ Запустите инференс на одном тестовом примере 🧪:
5️⃣.1️⃣ Для модели перевода с языка жестов на английский.
python demo_sign_language.py5️⃣.2️⃣ Для модели генерации рукописных цифр.
python demo_digits_generator.py --samples 9 --output_path ./generated_digits_grid.png