Your model contains one trainable layer which apparently should be used 4 times in a loop.
Since the number of output channels does not match the number on input channels, your code won’t work currently. Also you’ve forgot to return x in your forward method.