As other have rightly pointed out, verify you're using the Data Loader the right way. Ideally you need to create a custom dataset (in PyTorch terms) and apply all the transformations in this custom dataset. This might be helpful. Also, have you tried PyTorch Lightning?
btcmx
joined 2 years ago
For CV, Jetson family! The following source has a high level overview of challenges and solutions when deploying NVIDIA TAO on Jetson devices.
This post has a number of ML topics/skills you might (sooner or later) run into. In a sense it's kind of a roadmap/blueprint containing what you need to become an ML Engineer, (beyond learning a few topics/basics from a one(or more) courses.