use tensorflow estimator head api throws error

Question

I use head api as https://www.tensorflow.org/api_docs/python/tf/estimator/Head, but throws ValueError: trainable_variables cannot be None. Given None, here is the code

<