pytorch - 关于torchvision模型的时代数

Question

我试图找到，在 Imagenet 上训练的预训练 Alexnet 模型（可从 torchvision 获得）有多少个 epoch，以及使用的学习率是多少？我尝试检查检查点键以查看是否存储了任何时代信息。关于如何找到它的任何建议？

Answer 1

根据PyTorch 团队成员在 GitHub 上的评论，大部分培训都是使用https://github.com/pytorch/examples/tree/master/imagenet的变体完成的。所有模型都在 Imagenet 上进行了训练。根据文件：

• 默认学习率计划从 0.1 开始，每 30 个 epoch 衰减 10 倍，尽管他们建议 Alexnet 使用 0.01 作为初始学习率。
• epochs 的默认值为 90。