Aniket Patil
Aniket Patil
I also faced the same problem. - I would suggest you to check the **root data path** first. - Checked scaled images. As given by @JordanMakesMaps. - Use models from...
I am interested to take up this issue. Please assign this issue to me.
I'm willing to work on this issue and collaborate with the maintainers. Please let me know how I can proceed.
This is the code snippet: ```code def get_device(): return torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu') ``` I hope this will solve resolve your issues.