-
Notifications
You must be signed in to change notification settings - Fork 6
feat: add panama plus model #76
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
base: feat/add-classification-features-to-response
Are you sure you want to change the base?
Conversation
* Add new UKCEH models that have weights available (Kenya) --------- Co-authored-by: KatrionaGoldmann <kgoldmann@turing.ac.uk> Co-authored-by: Katriona Goldmann <katrionagoldmann@gmail.com>
trapdata/tests/test_pipeline.py
Outdated
|
|
||
| print(dir(SpeciesClassifierChoice)) | ||
|
|
||
| exit() |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Remove these when you are done debugging
trapdata/ml/models/localization.py
Outdated
|
|
||
| # Print the shape of each tensor in the state_dict | ||
| for name, param in state_dict.items(): | ||
| print(f"{name}: {param.shape}") |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Remove these when you are done troubleshooting. Also, instead of print statements you can drop into a interactive debugger by putting import pdb; pdb.set_trace() here, if that's helpful.
Added the PanamaPlus model for OOD detection