Комментарии 5
Это аналог pytorch-lighting/fast.ai или есть еще какие плюшки?
Lightning И Ignite выполняют примерно одни и те же задачи, одно из ключевых отличий, у Lightning есть возможность вызывать некоторые функции из коробки, которых нет в Ignite. Например обучение на нескольких графических процессорах. А у Fast.ai нет некоторых функций например преждевременной остановки обучения или сохранения контрольных точек.
Слои Linear и Dropout, нужны для того чтобы избежать переобучения.
Вероятно, тут какая-то ошибка. Дропаута в модели нет (или я не туда смотрю), а то, что полносвязный слой помогает избежать переобучения — это вообще какая-то фантастика.
Зарегистрируйтесь на Хабре, чтобы оставить комментарий
Сократить объем кода при помощи библиотеки PyTorch-Ignite