-
Notifications
You must be signed in to change notification settings - Fork 287
Open
Labels
bugSomething isn't workingSomething isn't working
Description
Search before asking
- I have searched the Pytorch-Wildlife issues and found no similar bug report.
Bug
The default version string for megadetectorv6, megadetectorv6-mit, megadetectorv6-apache and megadetectorv6-distributed are incorrect raising a value error when loading without passing a string.
Fix is to update each string in the constructor to a valid version to match the if else statements
Environment
No response
Minimal Reproducible Example
from PytorchWildlife.models import detection as pw_detection
# each call will fail
detection_model = pw_detection.MegaDetectorV6MIT(
device="cuda",
pretrained=True,
)
detection_model = pw_detection.MegaDetectorV6_Distributed(
device="cuda",
pretrained=True,
)
detection_model = pw_detection.MegaDetectorV6_Distributed(
device="cuda",
pretrained=True,
)
detection_model = pw_detection.MegaDetectorV6(
device="cuda",
pretrained=True,
)
Additional
No response
Are you willing to submit a PR?
- Yes I'd like to help by submitting a PR!
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
bugSomething isn't workingSomething isn't working
Type
Projects
Status
Backlog