ProxylessNAS yolo

  

   

    

     

    
    

     

      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)
     
    
   

    

     

    
    

     
 if not os.path.exists(cached_file) or overwrite:
     
    
   

    

     

    
    

     

       os.makedirs(model_dir, exist_ok=True)
     
    
   

    

     

    
    

     

       sys.stderr.write('Downloading: "{}" to {}\n'.format(url, cached_file))
     
    
   

    

     

    
    

     

       urlretrieve(url, cached_file)
     
    
   

    

     

    
    

     
 return cached_file
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      def load_url(url, model_dir='~/.torch/proxyless_nas', map_location=None):
     
    
   

    

     

    
    

     

       cached_file = download_url(url, model_dir)
     
    
   

    

     

    
    

     

       map_
     
    
  
 

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

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