I know you helped me solve this problem some time ago, but I have only managed to get back to it recently and I am not getting the results that I was expecting. So for:


self.base  outputs an activation of shape  [batch_size, 4] , so  out_labels  and  out_bbox should have  in_features=4


For this point do you mean that I should change to:
    self.out_labels = nn.Linear(4, nb_classes)
    self.out_bbox = nn.Linear(4, 4)

Because when I do that, I get the following error:
RuntimeError: size mismatch, m1: [4 x 314000], m2: [4 x 4] at /pytorch/aten/src/TH/generic/THTensorMath.cpp:940