Class output size from Model

[jaideep11061982]

1. What is the expected size of classouts

class_out, box_out = self.model(x)

I passed num_classes=4 , 0,1,2,3 where 0 is bg class.

My labels in dataset varry from 0 to 4 .

2. If i want to get class prob how i can get that. Model returns the class idx only.