pytorch-mobilenet-v2

MobileNetV2的PyTorch实现 这是MobileNetV2架构的PyTorch实现，如《》一文中所述。 *特别感谢@wangkuan为该模型提供了71.8％的top-1 acc！ 训练与准确性 要训​​练自己的模型，强烈建议使用小于0.5的辍学率以加快训练速度。 如本文所述，我尝试从头开始使用RMSprop训练模型，但是它似乎不起作用。 我目前正在使用SGD训练模型，并使其他超参数保持相同（但我使用的批次大小为256）。 将来，我还将尝试使用SGD检查点的RMSprop进行微调。 本文ImageNet的top-1准确性为71.7％ 。 我们目前的结果略高： 优化器 时代 前1-acc 预训练模型 RMSprop -- -- -- 新元 -- 71.8％ [ ] 新元+ RMSprop 去做 去做 去做 用法 要使用预训练的模型，请运行 from Mo