关于火炬视觉模型的时代数

Question

我试图找到在Imagenet上训练了多少个经过训练的Alexnet模型（可从torchvision获得）？还有学习率吗？ 我还尝试检查检查点键以查看是否存储了任何纪元信息。 关于如何找出答案的任何建议？ 我只是需要这些信息来进行比较研究。

致谢

Answer 1

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

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