Closed howardhsu closed 5 years ago
What is the expect memory to run train_autorec.py ? It seems to exhaust all memory to run this.
I got it. The loss needs to be detached.
Is this issue solved in the code ? I am also getting OOM while training the autoencoder.
What is the expect memory to run train_autorec.py ? It seems to exhaust all memory to run this.