Thank you so much for your answer! I wasn’t familiar with the difference between modules vs children - will try children instead
for the next part, I mean if only passing in one image, the Net won’t work and will return ’ [Error: Expected more than 1 value per channel when training]’