I was trying to implement this

  
      
      GitHub
  
  
    

hyqneuron/pytorch-avitm 3

pytorch-avitm - PyTorch implementation of AVITM


  
  
    
    
  
  


The error was while registering the parameters like this
self.logvar_bn .register_parameter(‘weight’, None)
self.mean_bn   .register_parameter(‘weight’, None)
self.decoder_bn.register_parameter(‘weight’, None)
self.decoder_bn.register_parameter(‘weight’, None)
in pytorch_model.py which was giving the errors.