张量流模型attention_ocr在使用gtx1070训练时出现oom错误

时间:2017-10-28 07:10:43

标签: tensorflow tensorflow-slim

我的gpu是gtx1070,当我想训练attention_ocr模型时,日志发生错误 ResourceExhaustedError:分配带形状的张量

时的OOM

文件说用gtx980进行了测试,所以理论上我的gtx1070还可以。

0 个答案:

没有答案