I am currently trying to port a TF 1.x project to TF version 2.x. I am running into trouble understanding how to port the TF 1 Estimators to TF 2. I create the same model as