d_marcos:

After investigation I realized my tensor would actually turn to None type after going through self.input_net in ConvNet forward.


That’s great debugging!
The reason for this is that you’ve forgotten to return out in the forward method of PreActResnetBlock. Adding it should solve the issue.