_____
This pull request includes
- [ ] Bug fix (non-breaking change which fixes an issue)
- [x] New feature (non-breaking change which adds functionality)
- [ ] Breaking change (fix or feature that would cause existing functionality to not work as expected)
- [ ] This change requires a documentation update
General description of your pull request with the list of new features and/or bugs.
run_id
without passingload_training
and Lightning module.weights
has highest priority. ifweights is None
, load model fromrun_id
is used.model = DetrR50(num_classes=91, weights="detr-r50")
load from downloaded weight
load_weights(model, weights="detr-r50")
load from .pth file
load_weights(model, weights="~/.aloception/weights/detr-r50/detr-r50.pth")
load from run_id
load_weights(model, run_id="your_run_id_heer", project_run_id="your_project_run_id",)