文件名称：pytorch-mobilenet-v2
文件大小：12.56MB
文件格式：ZIP
更新时间：2024-05-17 16:15:30
Python
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
【文件预览】：
pytorch-mobilenet-v2-master
----mobilenetv2_718.pth(13.54MB)
----__init__.py(0B)
----LICENSE(11KB)
----MobileNetV2.py(4KB)
----README.md(2KB)
----__pycache__()
--------dataset.cpython-36.pyc(3KB)
--------MobileNetV2.cpython-36.pyc(3KB)
----test.py(3KB)
----dataset.py(3KB)
----MobileNetV2.pyc(4KB)