在进行训练过程前,加上以下代码即可:
tf.config.experimental_run_functions_eagerly(True)
原因可参考该博文内容,和以下链接:
https://blog.csdn.net/weixin_43824178/article/details/99297237
https://www.tensorflow.org/guide/effective_tf2#use_tfconfigexperimental_run_functions_eagerly_when_debugging