Interestingly that doens’t work for me. Please look at the updated question, I added more info to the code. In that same Model class I also have a self.encoder atribute that is a ModulelList. Neither model.module.encoder nor model.module.znoise work.
I get AttributeError: ‘DataPrallel’ object has no attribute ‘znoise’.
I’m running wih 2 GPUs, if that matters!