Hi Thomas, thank you for your reply.
I don’t quite understand what you mean. Are you talking about the loss dictionary that is returned by the model() call? What do you mean with batchx num categories? How do I check the shape of the output, since the call is canceled before it returns.
The finetuning tutorial mentions that we should use num_categories + 1 (0 reserved for background) as the amount of classes for the model. I have 6 different classes, so I initialize my model with 7 classes. As class 0 is reserved, I have numbered my classes from 1 to 6.
Regards,
Kenneth