I do not know why - Githubissues

tensorflow / nmt

TensorFlow Neural Machine Translation Tutorial

Apache License 2.0

6.36k stars 1.96k forks source link

I do not know why #348

Closed wxs19920926 closed 6 years ago

wxs19920926 commented 6 years ago

ResourceExhaustedError: OOM when allocating tensor with shape[5776,500002] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc

gheorghelisca commented 6 years ago

I also have no idea :(

My setup is:

./native_client/deepspeech -h
...
TensorFlow: v1.6.0-18-g5021473
DeepSpeech: v0.2.0-alpha.7-10-gea21010

luozhouyang commented 6 years ago

Your tensor is too large, and the GPU has not enough memory.

I guess 500,002 is your vocab size, it's too large. You can try to reduce the size and try again.

wxs19920926 commented 6 years ago

thanks