ghostnet.pytorch:ImageNet上73.6％的GhostNet 1.0x预训练模型-源码

GhostNet的PyTorch实施 如GhostNet中所述重现GhostNet架构的由Kai Han，Yunhe Wang，祁天，郭建元，徐纯景，关于框架的ILSVRC2012基准。 预训练模型 建筑学 ＃ 参数 MFLOPs 前1名/前5名准确性（％） 5.181百万 140.77 73.636 / 91.228 from ghostnet import ghostnet net = ghostnet () net . load_state_dict ( torch . load ( 'pretrained/ghostnet_1x-9c40f966.pth' )) 培训策略 8个GPU上的批处理大小为1024 初学率0.4 重量衰减0.00004 辍学率0.2 BN重量无衰减 我们将上述设置保持不变，并使用以下不同的训练技术进行实验，以进行消融和繁殖。 在预热阶段，学