Input tensor ValueError when loading model BertClassifier Model

Question

--INTRO

I am training a BERT classifier using the google code from tf-models-official==2.3.0. Everything works like charm: training, evaluating, saving checkpoints.