Tharun
Tharun
Okay sure. I'll add that and update
I have made some changes. First it will check if cuda is availble or not. If cuda is available --> device = 'cuda' else --> device = 'cpu' and this...
> Changes looks good .. but you can try deleting unwanted files like py-cache and all And yes. I have deleted that directory. Sorry for the delay
pre-commit.ci autofix