'ScriptModel' object has no attribute 'save'

'ScriptModel' object has no attribute 'save'

原因：torch0.4.0没有保存功能

  

   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      class ScriptModel(torch.jit.ScriptModule):
     
    
   

    

     

    
    

     
 def __init__(self, N, M):
     
    
   

    

     

    
    

     

       super(ScriptModel, self).__init__()
     
    
   

    

     

    
    

     

       self.weight = torch.nn.Parameter(torch.rand(N, M))
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       self.conv1 = torch.nn.Conv2d(3, 64, kernel_size=7, stride=2, padding=3,
     
    
   

    

     

    
    

     

       bias=False)
     
    
   

    

     

    
    

     
 # self.bn1 = torch.nn.BatchNorm2d(64, momentum=0.1)
     
    
   

    

     

    
    

     

       self.bn1 = torch.nn.BatchNorm2d(64)
     
    
   

    

     

    
    

     

       self.relu = torch.nn.ReLU(inplace=True)
     
    
   

    

     

    
    

     

       self.maxpool = torch.nn.MaxPool2d(kernel_size=3, stride=2, padding=1)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       @torch.jit.script_method
     
    
   

    

     

    
    

     
 def forward(self, input):
     
    
   

    

     

    
    

     
 if input.sum() > 0:
     
    
   

    

     

    
    

     

       output = self.weight.mv(input)
     
    
   

    

     

    
    

     
 else:
     
    
   

    

     

    
    

     

       output = self.weight + input
     
    
   

    

     

    
    

     
 
     
    
  
 

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

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