Относительно количества эпох для моделей Torchvision

Question

Я пытался выяснить, сколько эпох было подготовлено для модели Alexnet (доступной в torchvision) на Imagenet? А также скорость обучения? Я также попытался проверить ключи контрольной точки, чтобы увидеть, была ли сохранена какая-либо информация об эпохе. Любые предложения о том, как это выяснить? Мне просто нужна была эта информация для сравнительного исследования.

Привет

Answer 1

Согласно этому комментарию на GitHub от члена команды PyTorch, большая часть обучения проводилась с вариантом https://github.com/pytorch/examples/tree/master/imagenet. Все модели были обучены на Imagenet. По файлу:

• График скорости обучения по умолчанию начинается с 0,1 и уменьшается в 10 раз каждые 30 эпох, хотя они рекомендуют использовать 0,01 для Alexnet в качестве начальной скорости обучения.
• Значение по умолчанию для эпох - 90.