Also, are the other layers such as self.RCNN_cls_score  and self.RCNN_bbox_pred in lines 243 and 245 respectively being trained? I mean, they  ideally should be, because otherwise, I don’t think we would have good results(please correct me if I am wrong). However, I got this doubt because these layers are not explicitly set to training mode in the def train() function.