Hi Konpat,
Thank you your advice very much!
I try your suggestion and add float() to all loss.item during iteration and face the same “out of memory error” at 20 epoch.
I’m using torch 0.4 now and seems there is no .detach() to be used… I feel puzzled on it, any suggestion is appreciated.
Thank you very much!
Best,
Peter