Для запуска обучения нейронной сети необходимо запустить команду python3 train.py. Чтобы пропустить обучение и приступить к классификации на уже обученной модели нужно запустить команду python3 classify.py, для этого необходим файл сохраненной модели формата model.pth Для запуска файлов без использования cuda ядер необходимо стереть\закоментировать ".cuda()" везде где он есть.
Во время обучения выводятся точность и потери тренировочных и валидационных данных, а также precision для контроля переобучения модели.