По количеству эпох для моделей torchvision

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


pytorch torchvision pre-trained-model
person himalayanZephyr    schedule 02.04.2019    source источник

Ответы (1)


arrow_upward
2
arrow_downward

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

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