https://github.com/linkyeu/efficientnet_pytorch/blob/master/efficientnet.py
需要torch1.10版本
最快的需要25ms gpu
https://github.com/zsef123/EfficientNets-PyTorch
还有这个:
https://github.com/jacke121/efficientnet-pytorch
1070上:efficientnet-b0
512 batch_seze 1需要25ms, 4需要56ms,batch_size为6就内存溢出
416 batch_seze 1需要25ms, 4需要56ms,batch_size为10就内存溢出
proxyless_gpu batch_size 20 可以,再大就不行了。