detecto
detecto copied to clipboard
core.Model object doesn't take pretrained argument
Describe the bug class detecto.core.Model(classes=None, device=None, pretrained=True)
Code and Data Share the code that caused the error as well as an example image/label from your dataset.
Environment:
- OS:
- Python version:
- Detecto version:
- torch version:
- torchvision version:
Additional context Add any other context about the problem here.
Be sure you're using the latest version: https://github.com/alankbi/detecto/releases/tag/v1.2.1