arnuhu08

Results 2 issues of arnuhu08

Can the torch version provided in the requirement be changed to the version?

Thank you for your work! I tried reproducing the results on your paper. Anytime I run the code it keeps returning train loss values as nan for the resnet-50 model...