ptrblck:

I’m not sure which difference you are pointing to.


torch.ones(1,1,256,256).to(self.device) is not created for “self.channel_1” before assignment while for “self.channel_2” and “self.channel_3” it is created before assignment.