Excessive CPU RAM being used by Pytorch even inside .cuda() mode
I am having issue with excessive CPU RAM usage with this coding even inside .cuda() mode
Could anyone advise ?

Problem is now solved using this github commit
