I have a TPUEstimator object and I want to know the total number of trained parameters.
How can I do?