Solved by simaiden in post #2 
              
              
                It’s because your class does not have those attributes but self.model. So you have to use model.model.conv1 and with others attributes as well