-
Notifications
You must be signed in to change notification settings - Fork 530
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
Update KD Notebook for classification #1595
Conversation
…how_to_use to the list of checked notebooks
# Conflicts: # tests/deci_core_unit_test_suite_runner.py
# Conflicts: # README.md
Fixed Cifar/Imagenet datasets that using torchvision transforms to support HasPreprocessingParams
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.
Just a single comment inline.
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.
Looks good already, I just have a small test sugestion
src/super_gradients/training/datasets/classification_datasets/torchvision_utils.py
Show resolved
Hide resolved
…' into feature/SG-1189-update-notebooks
…' into feature/SG-1189-update-notebooks
…esize implementation to match the implementation from Torchvision
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.
LGTM
# Conflicts: # Makefile
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.
LGTM
This PR update KD Notebook and fixes a few bugs related: