gau820827 / AI-writer_Data2Doc

PyTorch Implementation of NBA game summary generator.
Apache License 2.0
83 stars 22 forks source link

train problem #53

Open wanghechong opened 6 years ago

wanghechong commented 6 years ago

i come across a problem when i trian the model, when i have trained the model 90 minutes, the GPU suddenly report the error about out of memory, it shoule be noted that i use the pascal 12G by myself, no other one use the GPU at the same time? so i do not know where i make the mistake ?