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.

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?

For CV, Jetson family! The following source has a high level overview of challenges and solutions when deploying NVIDIA TAO on Jetson devices.