Hi, I found another issue - maybe related with Google Colab. When I switched TF version to 2.5.0 with
!pip install tensorflow==2.5.0
The epochs started be incredible slow - basically unusable. Time per one epoch is more than 25 mins, while before it takes 100s. I stopped my learning process because of that. If I switch my Google Colab Pro back to factory reset and not using tensorflow 2.5.0, it started to work fast as before, but after try to convert model, I got the same issue with InvalidArgumentError . Any workaround for this?