Tensorflow:AssertionError: Bad argument number for Name: 3, expecting 4 (bug)

1. 問題描述

LZ配置的tf環境,只要運行就會出現一堆類似上述的錯誤

AssertionError: Bad argument number for Name: 3, expecting 4 (bug)

WARNING: Entity <bound method Dense.call of <tensorflow.python.layers.core.Dense object at 0x7f4ab843b3c8>> 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_VERBOSITY=10) and attach the full output. Cause: converting <bound method Dense.call of <tensorflow.python.layers.core.Dense object at 0x7f4ab843b3c8>>: AssertionError: Bad argument number for Name: 3, expecting 4

2. 解決方案

因爲對結果的影響不可觀,所以就沒去在意 ,後面發現用其他docker並沒有多少問題,而且每次都出現一堆warning很影響美觀性,於是百度準備解決這個問題

後來發現是有個gast的庫版本太高,導致不兼容的問題,降級gast即可解決

使用pip或者conda進行降級

pip install --user gast==0.2.2

conda install gast==0.2.2

3. 參考地址

1. https://github.com/tensorflow/autograph/issues/1