Training on jetson-inference getting error FileNotFoundError: Found no valid file for the classes Drowsy. Supported extensions are: .jpg, .jpeg,

I solved the problem by uninstalling pytorch and torchvision and installing the bundled pytorch that came with the jetson-inference tools.