second time backward error

[Neronjust2017]

Hi Fengchun,

After I download and process MNIST dataset and run "main_Digits.py", I got the following error: "RuntimeError: Trying to backward through the graph a second time, but the buffers have already been freed. Specify retain_graph=True when calling backward the first time". Could tell me what happens here? And when I specify retain_graph=True, this error dispeared. Is this an error in your code? or someting else caused this error?

Thanks!

[joffery]

Hi @Neronjust2017,

Thanks for pointing out it.
The training of WAE should be the same as https://github.com/1Konny/WAE-pytorch, where "retain_graph=True " is necessary. I accidentally deleted it when cleaning the code.

Best,
Fengchun