Hi, that solution works, thanks! I’m now trying to modify the x and b parameters from this ShiftLayer based on some conditions
netG.x and netG.b does not work, it gives me the error
torch.nn.modules.module.ModuleAttributeError: 'Generator' object has no attribute 'x'


So I tried accessing them through netG.module.x but that does not work either and I get the error:
torch.nn.modules.module.ModuleAttributeError: 'Generator' object has no attribute 'module'


How can I do this?