关于 torchvision 模型的纪元数
Regarding number of epochs for torchvision models
我想知道，预训练的 Alexnet 模型（可从 torchvision 获得）在 Imagenet 上训练了多少个 epoch，以及使用了多少学习率？我尝试检查检查点密钥以查看是否存储了任何纪元信息。
关于如何找到它有什么建议吗？
根据this comment on GitHub by a PyTorch team member, most of the training was done with a variant of https://github.com/pytorch/examples/tree/master/imagenet。所有模型都在 Imagenet 上训练。根据文件：
- 默认学习率计划从 0.1 开始，每 30 个时期衰减 10 倍，尽管他们建议对 Alexnet 使用 0.01 作为初始学习率。
- 纪元的默认值为 90。
我想知道，预训练的 Alexnet 模型（可从 torchvision 获得）在 Imagenet 上训练了多少个 epoch，以及使用了多少学习率？我尝试检查检查点密钥以查看是否存储了任何纪元信息。 关于如何找到它有什么建议吗？
根据this comment on GitHub by a PyTorch team member, most of the training was done with a variant of https://github.com/pytorch/examples/tree/master/imagenet。所有模型都在 Imagenet 上训练。根据文件：
- 默认学习率计划从 0.1 开始，每 30 个时期衰减 10 倍，尽管他们建议对 Alexnet 使用 0.01 作为初始学习率。
- 纪元的默认值为 90。