Tensorflow Tacotron2模型混合精度训练错误

系统信息:

GPU类型:Tesla T4 Nvidia驱动程序版本:418.87.01 CUDA版本:10.1.243 CUDNN版本:7.6.3 Python版本(如果适用):3.7.4 TensorFlow版本(如果适用):1.14.0 操作系统+版本:Ubuntu 16.04.6 LTS(GNU / Linux 4.4.0-142-generic x86_64)

您好,我正在使用混合精度训练来训练Tensorflow版本Tacotron2模型。 经过一些训练迭代后,会发生错误,详细的错误信息如下:

Exiting due to exception: 2 root error(s) found. (0) Invalid argument: TensorArray dtype is float but Op is trying to write dtype half. [[node Tacotron_model/inference/encoder_LSTM/bidirectional_rnn/fw/fw/TensorArrayUnstack/TensorArrayScatter/TensorArrayScatterV3 (defined at /home/yichao.li/lite-tacotron2/tacotron/models/modules.py:225) ]] [[strided_slice_51/_7343]] (1) Invalid argument: TensorArray dtype is float but Op is trying to write dtype half. [[node Tacotron_model/inference/encoder_LSTM/bidirectional_rnn/fw/fw/TensorArrayUnstack/TensorArrayScatter/TensorArrayScatterV3 (defined at /home/yichao.li/lite-tacotron2/tacotron/models/modules.py:225) ]] 0 successful operations. 0 derived errors ignored.

您能帮忙澄清一下吗?如何解决这个问题?

yx118226 回答:Tensorflow Tacotron2模型混合精度训练错误

暂时没有好的解决方案,如果你有好的解决方案,请发邮件至:iooj@foxmail.com
本文链接:https://www.f2er.com/3160886.html

大家都在问