Hi,
Your model definition is correct and its output has the shape of [batch_size, 113]:
x = torch.randn(12, 3, 120, 120)
Net()(x).shape

Could you explain how did you print this values? in which line?



 Jishan:

torch.Size([48, 133])


Also, I think error is from somewhere else not the model. Could you share the stacktrace error?
Bests