cuda memory out - Githubissues

edenton / svg

186 stars 56 forks source link

cuda memory out #5

Open wyhsirius opened 5 years ago

wyhsirius commented 5 years ago

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 commented 5 years ago

I have the same issue :(

fanshuhuangjia commented 3 years ago

change the batch size,the default batch size is 100