Quick question, when I assign ‘output = self.conv(x_conv)’, is this output going to be stored on the right device? Do I have to do any sort of specification when performing the convolution? Also, do you see anywhere I may need to add ‘device=x.device’ in the forward other than under_sum and normalized_channels? Thanks once again.