-
Notifications
You must be signed in to change notification settings - Fork 484
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
Feature/sg 631 add registry collate funcs samplers #668
Feature/sg 631 add registry collate funcs samplers #668
Conversation
added all_collate_functions.py to training.datasets for registy
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.
minor comment
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.
Comment regarding more collate functions left out, inline.
More importantly, the fact you added a registry for collate functions iis great but not sufficient.
Suppose a user registers their collate functions, and we pass some string as an argument to train_dataloader_params. Will that string translate to the right collate function?
The answer is no. You need to resolve the right class by implementing a factory otherwise the registry is useless.
Guidance: see dataloaders.get(), find where we handle the dataloader_params processing and resolve the collate_fn argument with your new CollateFactory.
…ttps://github.com/Deci-AI/super-gradients into feature/SG-631_add_registry_collate_funcs_samplers
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.
Almost there, just 2 more comments 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.
LGTM
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
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.
first code contribution to SG - Good luck!
No description provided.