Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

paddle lstm cpu版本测试出错 #60

Open
leanna62 opened this issue Jan 16, 2018 · 0 comments
Open

paddle lstm cpu版本测试出错 #60

leanna62 opened this issue Jan 16, 2018 · 0 comments

Comments

@leanna62
Copy link
Contributor

镜像版本:dzhwinter/benchmark:latest;
CPU版本测试,环境变量设置
export CUDA_VISIBLE_DEVICES=;
export OMP_NUM_THREADS=1
运行命令:python stacked_dynamic_lstm.py --device CPU
输出信息如下:
Reading data to memory
Done. data size 24996
Traceback (most recent call last):
File "stacked_dynamic_lstm.py", line 206, in
main()
File "stacked_dynamic_lstm.py", line 187, in main
train_loop(args.pass_num, args.crop_size)
File "stacked_dynamic_lstm.py", line 174, in train_loop
tensor_words = to_lodtensor([x[0] for x in data], place)
File "stacked_dynamic_lstm.py", line 201, in to_lodtensor
res.set_lod([lod])
MemoryError: std::bad_alloc

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant