• Home
• uk.ac.gate.plugins
• learningframework
• 4.1.1
• source code
• gatelfpytorchjson.CustomModule.html

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

resources.wrappers.FileJsonPyTorch.gate-lf-pytorch-json.docs.pythondoc.gatelfpytorchjson.CustomModule.html Maven / Gradle / Ivy

  
    
    
    
    
    
    
    
    
    
    
    
    
    
    
    
     
  
  
      

    

      

        

          

            
  

gatelfpytorchjson.CustomModule module¶


class gatelfpytorchjson.CustomModule.CustomModule(config={})[source]¶
Bases: torch.nn.modules.module.Module


get_lossfunction(config={})[source]¶




get_optimizer(config={})[source]¶




on_cuda()[source]¶
Returns true or false depending on if the module is on cuda or not. Unfortunately
there is no API method in PyTorch for this so we get this from the first parameter of the
model and cache it.
NOTE: this must be called outside of the init() method, because the cuda status of the module
gets set by the modelwrapper.




set_seed(seed)[source]¶