-
Notifications
You must be signed in to change notification settings - Fork 622
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
Add a typical data loading pipeline path for the EfficeintNet #5761
base: main
Are you sure you want to change the base?
Conversation
- adds an option to run the EfficeintNet network with a typical data loading pipeline without very advanced optimization that most users won't implement Signed-off-by: Janusz Lisiecki <[email protected]>
!build |
docs/examples/use_cases/pytorch/efficientnet/image_classification/dataloaders.py
Fixed
Show fixed
Hide fixed
docs/examples/use_cases/pytorch/efficientnet/image_classification/dataloaders.py
Fixed
Show fixed
Hide fixed
CI MESSAGE: [21708341]: BUILD STARTED |
input = next_input | ||
target = next_target | ||
def prefetched_loader(loader, num_classes, one_hot, typical_loader ): | ||
if typical_loader: |
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.
I would consider building a TypicalPytorchLoader as a separate class that just omits prefetching and fast_collate instead of introducing if-branches everywhere. It can even land in a separate file with docstring explaining that it's created according to the doc.
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.
Fixed
Signed-off-by: Janusz Lisiecki <[email protected]>
f856c27
to
fffd337
Compare
!build |
CI MESSAGE: [21709902]: BUILD STARTED |
CI MESSAGE: [21709902]: BUILD PASSED |
CI MESSAGE: [21781968]: BUILD STARTED |
CI MESSAGE: [21785497]: BUILD STARTED |
Signed-off-by: Janusz Lisiecki <[email protected]>
CI MESSAGE: [21790097]: BUILD STARTED |
CI MESSAGE: [21785497]: BUILD PASSED |
CI MESSAGE: [21790097]: BUILD PASSED |
data loading pipeline without very advanced optimization that
most users won't implement
Category:
Other (e.g. Documentation, Tests, Configuration)
Description:
data loading pipeline without very advanced optimization that
most users won't implement
Additional information:
Affected modules and functionalities:
Key points relevant for the review:
Tests:
Checklist
Documentation
DALI team only
Requirements
REQ IDs: N/A
JIRA TASK: N/A