ShuffleNetV2-pytorch

ShuffleNetV2-pytorch copied to clipboard

Extremely slow data load when training on ImageNet

1

Thanks for your excellent work. I found that the data load speed is very slow when training the shuffleNetV2-0.5X. The GPU is basically not working. I use one SSD and...

gasvn

Hi, @Randl I met this error when run "python3 imagenet.py --dataroot "pl_5plus3/" --resume "model_best.pth.tar" -e --scaling 0.5". ... ) number of parameters: 1372120 FLOPs: 40376352.0 => loading checkpoint 'model_best.pth.tar' Traceback...

bemoregt

Hello! Thanks a lot. Do you also plan to maybe include an AutoShuffle version of the ShuffleNetV2? https://arxiv.org/abs/1901.08624

kayuksel

Would you please share the shufflenet v2 x1.0 pretrained model? @Randl

ArsenLuca

Hi, i have a problem. I noticed the self.n[3, 8, 3] in line 185 of model.py, but the third stage of n is 7 in the paper. Maybe self.n[3, 7,...

mosjoker