LightModels icon indicating copy to clipboard operation
LightModels copied to clipboard

the pytorch implementation of light models: SqueezeNet, ShuffleNet, MobileNet, MobileNetv2 and ShuffleNetv2

Results 4 LightModels issues
Sort by recently updated
recently updated
newest added

How do I change to my own data set,which are csv files?

论文里在shufflenet v1中的bottleneck中应该使用了两次relu吧,应该只有第一次的PWconv和shortcut之后使用了激活函数,代码里两个PWconv和DWconv都使用了relu。

in the method :test of main.py ,net.load_state_dict('./checkpoint/{}.pth'.format(args.model)) cannot run ,but net.load_state_dict(torch.loda('./checkpoint/{}.pth'.format(args.model))) can solve this problem.

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