roboflow / notebooks

Examples and tutorials on using SOTA computer vision models and techniques. Learn everything from old-school ResNet, through YOLO and object-detection transformers like DETR, to the latest models like Grounding DINO and SAM.

https://roboflow.com/models

5.43k stars 855 forks source link

update train-huggingface-detr-on-custom-dataset.ipynb to work with pytorch-lightning >= 2.0.0 #85

Closed SkalskiP closed 1 year ago

SkalskiP commented 1 year ago

Description

update train-huggingface-detr-on-custom-dataset.ipynb to work with pytorch-lightning >= 2.0.0

List any dependencies that are required for this change.

Type of change

Please delete options that are not relevant.

[x] Bug fix (non-breaking change which fixes an issue)
[ ] New feature (non-breaking change which adds functionality)
[ ] This change requires a documentation update