36 lines
2.2 KiB
Markdown
36 lines
2.2 KiB
Markdown
# Инструкция по запуску
|
|
- Запускаем контейнер и подключаемся к нему
|
|
- Устанавливаем проект (тащим с гита, переносим любым удобным способом)
|
|
- Включаем venv
|
|
- Дотягиваем зависимости
|
|
- Запускаем как `python3 train.py`
|
|
|
|
## train.py — Основной скрипт для обучения модели
|
|
Этот файл запускает процесс обучения, тестирования и, возможно, сохранения модели. Он объединяет остальные компоненты проекта.
|
|
|
|
Содержимое:
|
|
- Импорты всех необходимых модулей (модель, данные, утилиты, конфигурации).
|
|
- Чтение гиперпараметров из config.yaml.
|
|
- Основной цикл обучения.
|
|
- Тестирование модели на валидационных данных.
|
|
|
|
## model.py — Определение архитектуры модели
|
|
Этот файл содержит только архитектуру модели. Если необходимо сменить модель, достаточно будет изменить только этот файл.
|
|
|
|
Содержимое:
|
|
- Определение слоёв и функции forward для прохождения данных через модель.
|
|
|
|
## datasets.py — Работа с данными
|
|
Этот файл отвечает за загрузку, подготовку и трансформацию данных.
|
|
|
|
Содержимое:
|
|
- Загрузка тренировочных и тестовых наборов данных.
|
|
- Определение необходимых трансформаций.
|
|
|
|
## utils.py — Вспомогательные функции
|
|
Этот файл содержит функции, которые помогают сократить основной код и избежать дублирования.
|
|
|
|
Содержимое:
|
|
- Сохранение/загрузка модели.
|
|
- Подсчёт точности.
|