cuda memory out

[wyhsirius]

Hello, thanks for your work. However, when I start to train your KTH model, I have 'cuda memory out' problem. Could you please explain what kind of training setting you used ? Have you trained on multiple GPUs ?

[malalejandra]

I have the same issue :(

[fanshuhuangjia]

change the batch size,the default batch size is 100