Tornike:

RuntimeError: The size of tensor a (380) must match the size of tensor b (3) at non-singleton dimension 0


If I’m not mistaken, in your model’s forward pass in rcnn module self.transform trying to normalize image, but instead of image with 3 channels it gets 380