alphafold_casp13 tensorflow.python.layers.normalization.BatchNormalization AssertionError: Bad argument number for Name: 3, expecting 4

[truatpasteurdotfr]

running the plain cpu version I get a lot of:

WARNING:tensorflow:Entity <bound method BatchNormalization.call of <tensorflow.python.layers.normalization.BatchNormalization object at 0x7efb6e190940>> 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 BatchNormalization.call of <tensorflow.python.layers.normalization.BatchNormalization object at 0x7efb6e190940>>: AssertionError: Bad argument number for Name: 3, expecting 4

[Augustin-Zidek]

Could you verify if installing gast resolves the issue?

pip install --user gast==0.2.2

[RubensZimbres]

Same issue here although the rest of the code runs:

WARNING:tensorflow:Entity <bound method BatchNormalization.call of <tensorflow.python.layers.normalization.BatchNormalization object at 0x7f8717088208>> 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 BatchNormalization.call of <tensorflow.python.layers.normalization.BatchNormalization object at 0x7f8717088208>>: AssertionError: Bad argument number for Name: 3, expecting 4 I0306 02:18:36.437770 140705132992256 secstruct.py:58] Creating secstruct Tensor("competitionsep/contacts_net/concat_2:0", shape=(?, 128, 256), dtype=float32, device=/device:CPU:0)

In Anaconda environment with Debian + PyTorch + CUDA set up on a GCP instance, pip install --user gast==0.2.2 didn't solve the problem

[truatpasteurdotfr]

thanks adding gast==0.2.2 fixed the issue (both cpu and gpu version). cpu only version:

(alphafold-cpu) [tru@ld19-1020 alphafold-cpu]$ python3 -m pip list
Package                Version    
---------------------- -----------
absl-py                0.8.1      
astor                  0.8.1      
cloudpickle            1.3.0      
contextlib2            0.6.0.post1
decorator              4.4.2      
dm-sonnet              1.35       
gast                   0.2.2      
google-pasta           0.1.8      
grpcio                 1.27.2     
h5py                   2.10.0     
Keras-Applications     1.0.8      
Keras-Preprocessing    1.1.0      
Markdown               3.2.1      
numpy                  1.16.0     
pip                    20.0.2     
protobuf               3.11.3     
semantic-version       2.8.4      
setuptools             41.0.0     
six                    1.12.0     
tensorboard            1.14.0     
tensorflow             1.14.0     
tensorflow-estimator   1.14.0     
tensorflow-probability 0.7.0      
termcolor              1.1.0      
Werkzeug               1.0.0      
wheel                  0.34.2     
wrapt                  1.12.0     

gpu version:

(alphafold) [tru@ld19-1020 alphafold]$ python3 -m pip list
Package                Version    
---------------------- -----------
absl-py                0.8.1      
astor                  0.8.1      
cloudpickle            1.3.0      
contextlib2            0.6.0.post1
decorator              4.4.1      
dm-sonnet              1.35       
gast                   0.2.2      
google-pasta           0.1.8      
grpcio                 1.27.2     
h5py                   2.10.0     
Keras-Applications     1.0.8      
Keras-Preprocessing    1.1.0      
Markdown               3.2.1      
numpy                  1.16.0     
pip                    20.0.2     
protobuf               3.11.3     
semantic-version       2.8.4      
setuptools             41.0.0     
six                    1.12.0     
tensorboard            1.14.0     
tensorflow             1.14.0     
tensorflow-estimator   1.14.0     
tensorflow-gpu         1.14.0     
tensorflow-probability 0.7.0      
termcolor              1.1.0      
Werkzeug               1.0.0      
wheel                  0.34.2     
wrapt                  1.12.0