CancelledError 这错误难道跟显卡有关?
来源:7-12 subword文本分类之dataset变换与模型训练
AY9717
2019-12-28
我在 fit的时候,第一次成功了,然后第二次报出了这样的错误
CancelledError: [Derived]RecvAsync is cancelled.
[[{{node Adam/Adam/update/AssignSubVariableOp/_41}}]]
[[Reshape_11/_38]] [Op:__inference_distributed_function_6316]
Function call stack:
distributed_function
查了半天说是跟显卡有关,所以特来求证。
写回答
1回答
-
对,貌似只跟GPU上的LSTM有关,参考:https://blog.csdn.net/qq_36659192/article/details/103198825
可以设置GPU auto growth来避免这个问题。
012020-01-03
相似问题