Model quantization aware training problem

General Discussion
,
#1

I’m getting this error message while training sub-model “module ‘tensorflow.keras.layers’ has no attribute 'MultiHeadAttention”.

#3

does this address your problem? Module 'tensorflow.keras.layers' has no attribute 'MultiHeadAttention' · Issue #9357 · tensorflow/models · GitHub