WARNING:tensorflow:Entity bound method BatchNormalization.call of

信息描述

在使用TensorFlow的時候報一堆的warning

WARNING:tensorflow:Entity <bound method BatchNormalization.call of <tensorflow.python.layers.normalization.Ba$
chNormalization object at 0x7f3905978cf8>> could not be transformed and will be executed as-is. Please report
this to the AutgoGraph team. When filing the bug, set the verbosity to 10 (on Linux, `export AUTOGRAPH_VERBOS$
TY=10`) and attach the full output. Cause: converting <bound method BatchNormalization.call of <tensorflow.pyt
hon.layers.normalization.BatchNormalization object at 0x7f3905978cf8>>: AssertionError: Bad argument number fo
r Name: 3, expecting 4

環境

•  TensorFlow：1.14

• python：3.16

解決辦法

安裝gast庫

pip install gast==0.2.2

安裝之後，這些warning就消失了

參考：https://github.com/youdao-ai/SRNet/issues/4