Error During Training：AttributeError: 'SSODROIHeads' object has no attribute 'get_preds_for_pseudo_labels'

[ysysys666]

Thank you for your excellent work!
When I tried to reproduce your experiment, I found an "AttributeError: 'SSODROIHeads' object has no attribute 'get_preds_for_pseudo_labels' ", The error location is at line 76 in custom_rcnn.py. preds_on_gt = self.roi_heads.get_preds_for_pseudo_labels(weak_features, [x.t_boxes for x in gt_instances]), I would like to ask you about it.

[wkfdb]

i checked the code but without running, the function in line 76 in custom_rcnn.py has been changed to get_preds_from_weak_features. It should work. Pls tell me if new error occurs, I will find time to check it by running.

[ysysys666]

Thank you very much! And I still want to consult why "RuntimeError: DataLoader worker (pid 41346) is killed by signal: Aborted." occurs after training?