train_autorec.py exhausts all my memory.

RaymondLi0 / conversational-recommendations

70 stars 21 forks source link

Closed howardhsu closed 5 years ago

howardhsu commented 5 years ago

What is the expect memory to run train_autorec.py ? It seems to exhaust all memory to run this.

howardhsu commented 5 years ago

I got it. The loss needs to be detached.

malzantot commented 4 years ago

Is this issue solved in the code ? I am also getting OOM while training the autoencoder.