Skip to content

paddle lstm cpu版本测试出错 #60

@leanna62

Description

@leanna62

镜像版本: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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions