【问题标题】:Regarding number of epochs for torchvision models关于torchvision模型的时代数
【发布时间】:2019-04-02 14:14:25
【问题描述】:

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

【问题讨论】:

    标签: pytorch pre-trained-model torchvision


    【解决方案1】:

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

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

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 1970-01-01
      • 2021-05-25
      • 2021-11-30
      • 2020-12-05
      • 2019-03-09
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      相关资源
      最近更新 更多