pytorch/vision
View on GitHubImprove error message when number of anchors mismatch
Open
#1,539 opened on Oct 30, 2019
enhancementhelp wantedmodule: modelstopic: object detection
Repository metrics
- Stars
- (15,050 stars)
- PR merge metrics
- (Avg merge 12d 8h) (14 merged PRs in 30d)
Description
See discussion in https://github.com/pytorch/vision/issues/1308#issuecomment-547964507
We should add checks in https://github.com/pytorch/vision/blob/067704986d92a3dff4ccf4a686080b3d31390402/torchvision/models/detection/rpn.py#L440 so that it's clearer to users what they need to change if they face such an error in the future.