【发布时间】:2019-04-02 14:14:25
【问题描述】:
我试图找出,预训练的 Alexnet 模型(可从 torchvision 获得)在 Imagenet 上训练了多少个 epoch,以及使用的学习率是多少?我尝试检查检查点键以查看是否存储了任何时代信息。 有关如何找到它的任何建议?
【问题讨论】:
标签: pytorch pre-trained-model torchvision
我试图找出,预训练的 Alexnet 模型(可从 torchvision 获得)在 Imagenet 上训练了多少个 epoch,以及使用的学习率是多少?我尝试检查检查点键以查看是否存储了任何时代信息。 有关如何找到它的任何建议?
【问题讨论】:
标签: pytorch pre-trained-model torchvision
根据 PyTorch 团队成员在 GitHub 上的 this comment,大部分培训都是使用 https://github.com/pytorch/examples/tree/master/imagenet 的变体完成的。所有模型都在 Imagenet 上进行了训练。根据文件:
【讨论】: