Solved by avalon1511 in post #7 
              
              
                Thanks for clarifying! Following your advice I  was getting an error trying to do what you suggested: 
AttributeError: 'Generator' object has no attribute 'module'

And when I printed my state_dict() using print(Generator.state_dict()) I got this: 
 ('main.10.weight', tensor(<size>),  device='cuda:0…