train.py error: ResourceExhaustedError when I run train.py for dataset from download.sh

Kyubyong / transformer

A TensorFlow Implementation of the Transformer: Attention Is All You Need

Apache License 2.0

4.28k stars 1.3k forks source link

Open DuyenPay opened 5 years ago

DuyenPay commented 5 years ago

2019-04-17_135655 This is the error description. How can I fix this?

trx14 commented 5 years ago

You could try small batch size or block. It seems that your GPU capacity is not enough.

yifanjun233 commented 5 years ago

i met this problem too and i found the memory-usage is about full but the GPU-util is zero.have you met this?