This might be related to InvalidArgumentError: required broadcastable shapes [Op:Mul] 8. Are you training your model inside Google Colab? If so, have you fixed the Tensorflow version, or are you using the default version used by Colab?