Проект: PyTorch
Описание: проект для развития навыка использования возможностей PyTorch. Первоначально это маленький помощник, который говорит мне, стоит ли пить кофе. Вероятно, это будет дополняться печеньками и полноценными завтраками.
Что делать, чтоб тебе ответили.
- Есть файл requirements.txt, чтобы подтянуть библиотечки.
- Весь процесс осуществляется в папке code/
- Там есть уже готовая модель (coffee_model.pth) для выполнения кода main.py. Поэтому можно сразу его запустить, в консоли ответить на 4 вопроса (1=да, 0=нет) и получить ответ. Без готовой модели запуск main.py "покраснеет". Убедитесь, что кофейная модель рядом с main.py.
- Можно сначала запустить файл learning.py для обучения "нейросети". Файл coffee_model.pth будет создан или перезапишется, если существует.
- Если есть графический движок Nvidia, то обучение запустится на нём. Если нет, то на ЦП.
- Спасибо за ваше время <3