[[node gradient_tape/model/multi_head_attention_1/multi_head_attention_1-Attention/truediv_1/RealDiv (defined at D:/pythonPro66/att_mtcn_cli_train.py:427) ]] Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info. [Op:__inference_train_function_6511]
时间: 2024-04-05 17:30:49 浏览: 148
这看起来像是一个 TensorFlow 报错信息,提示了一个节点的定义位置以及建议的解决方法。这个错误信息表明在训练过程中,模型在执行某个操作时内存溢出了,导致程序崩溃。建议你按照提示添加 `report_tensor_allocations_upon_oom` 选项,这样能够在程序内存溢出时输出一份当前内存分配情况的报告。这样可以更好地帮助你查找问题所在并进行调试。
阅读全文