zh-NER-TF icon indicating copy to clipboard operation
zh-NER-TF copied to clipboard

Published 20 hours ago • verified publisher icon Determined22

训练时内存持续增长到OOM

Open xiaofei05 opened this issue 5 years ago • 2 comments

用的自己的数据，数据量比较大， 训练的时候大概batch_size=128 2000 step时出现OOM，请问是哪里出现问题了呢

xiaofei05 avatar Apr 20 '19 14:04 xiaofei05

同出现此问题

GanjinZero avatar May 23 '19 06:05 GanjinZero

把main.py中的这一行改了一下

config.gpu_options.per_process_gpu_memory_fraction = 0.9

就好了

GanjinZero avatar May 24 '19 01:05 GanjinZero