Инструкция по запуску

• Запускаем контейнер и подключаемся к нему
• Устанавливаем проект (тащим с гита, переносим любым удобным способом)
• Включаем venv
• Дотягиваем зависимости
• Запускаем как python3 train.py

train.py — Основной скрипт для обучения модели

Этот файл запускает процесс обучения, тестирования и, возможно, сохранения модели. Он объединяет остальные компоненты проекта.

Содержимое:

• Импорты всех необходимых модулей (модель, данные, утилиты, конфигурации).
• Чтение гиперпараметров из config.yaml.
• Основной цикл обучения.
• Тестирование модели на валидационных данных.

model.py — Определение архитектуры модели

Этот файл содержит только архитектуру модели. Если необходимо сменить модель, достаточно будет изменить только этот файл.

Содержимое:

• Определение слоёв и функции forward для прохождения данных через модель.

datasets.py — Работа с данными

Этот файл отвечает за загрузку, подготовку и трансформацию данных.

Содержимое:

• Загрузка тренировочных и тестовых наборов данных.
• Определение необходимых трансформаций.

utils.py — Вспомогательные функции

Этот файл содержит функции, которые помогают сократить основной код и избежать дублирования.

Содержимое:

• Сохранение/загрузка модели.
• Подсчёт точности.