EfficientNet 测试

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 可以，再大就不行了。
 

  

   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      def test():
     
    
   

    

     

    
    

     

       x = torch.FloatTensor(4, 3, 512, 512).cuda()
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       w, d, _, p = efficientnet_params('efficientnet-b0')
     
    
   

    

     

    
    

     
 # note: all models have drop conn
     
    
  
 

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/90708836