-
Notifications
You must be signed in to change notification settings - Fork 7.2k
Closed
Description
As noted in #5209 (comment)
It would be great if these can be replaced by ValueErrors.
https://github.com/pytorch/vision/blob/main/torchvision/models/detection/backbone_utils.py#L114
https://github.com/pytorch/vision/blob/main/torchvision/models/detection/backbone_utils.py#L127
Maybe these too?
https://github.com/pytorch/vision/blob/main/torchvision/models/detection/backbone_utils.py#L155
https://github.com/pytorch/vision/blob/main/torchvision/models/detection/backbone_utils.py#L186
Once the assert is replaced by valueError. We should add test to check for it.
Probably here
https://github.com/pytorch/vision/blob/main/test/test_backbone_utils.py
abhi-glitchhg