文件名称:regnet.pytorch:具有一系列预训练模型的PyTorch风格和人类可读的RegNet
文件大小:9KB
文件格式:ZIP
更新时间:2024-05-24 12:52:07
pytorch imagenet pretrained-models resnext neural-architecture-search
使用TorchVision样式的RegNet实施 Ilja Radosavovic,Raj Prateek Kosaraju,Ross Girshick,Kaiming He和PiotrDollár的PyTorch实现。 与相比,此存储库遵循样式,预计该样式可通过预先存在的下游应用程序更轻松地解释和使用。 我们在配备12G VRAM的8个TITAN XP GPU上训练以下模型。 在前五个时期中,我们将学习率从0.1线性提高。 预训练模型 模型 参数(M) GFLOPs 批量大小 前1个帐户(%)(我们的实现) 前1个acc(%)(官方) 2.685 0.199 1024 68.210 68.9 5.158 0.398 1024 72.278 72.7 6.196 0.601 1024 73.862 74.1 7.260 0.800 1024
【文件预览】:
regnet.pytorch-master
----regnet.py(7KB)
----LICENSE(1KB)
----README.md(4KB)
----.gitignore(2KB)
----regnet_fb.py(10KB)