SawantMyelin

Thanks for the response, let me check and i will let you know.

I got it to work by removing data parallelism line and doing some other modifications in the main.py and removing all the .cuda() lines in other code. The results were...