MemoryError
Lorne0 opened this issue · 1 comments
Lorne0 commented
I tried to implement it on GPU without any revise of the hyperparameters(only the path). However, I got
MemoryError: Error allocating 25690112 bytes of device memory (out of memory).
when it was on epoch 2.
I'm not sure if I use wrong command:
THEANO_FLAGS='mode=FAST_RUN,floatX=float32,device=gpu0' python evaluate_flickr30k.py
It shows CNMeM is disabled.
Then I try:
THEANO_FLAGS='mode=FAST_RUN,floatX=float32,device=gpu0,lib.cnmem=1' python evaluate_flickr30k.py
However, when it runs gen_sample, the sample sentence is vanished, which is really strange.
Or something wrong with the preprocessing?
I make the four pkl files by
#1
Lorne0 commented
After use intuinno's code this problem is disappeared.