ProxylessNAS pytorch

原文：https://github.com/mit-han-lab/ProxylessNAS/blob/master/proxyless_nas/model_zoo.py

proxyless_gpu
 

1070 416 8 36ms batch_size最高为20

proxyless_mobile_14
 

1070 416 8 51ms

  

   

    

     

    
    

     

      import time
     
    
   

    

     

    
    

     

      from collections import OrderedDict
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import math
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import torch.nn as nn
     
    
   

    

     

    
    

     

      import torch.optim
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import os
     
    
   

    

     

    
    

     

      import sys
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      from functools import partial
     
    
   

    

     

    
    

     

      import json
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      try:
     
    
   

    

     

    
    

     
 from urllib import urlretrieve
     
    
   

    

     

    
    

     

      except ImportError:
     
    
   

    

     

    
    

     
 from urllib.request import urlretrieve
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      def download_url(url, model_dir="~/.torch/proxyless_nas", overwrite=False):
     
    
   

    

     

    
    

     

       model_dir = os.path.expanduser(model_dir)
     
    
   

    

     

    
    

     

       filename = url.split('/')[-1]
     
    
   

    

     

    
    

     

       cached_file = os.path.join(model_dir, filename)
     
    
  
 

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

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