TPU estimator does not shutdown

tensorflow / tpu

Reference models and tools for Cloud TPUs.

https://cloud.google.com/tpu/

Apache License 2.0

5.22k stars 1.77k forks source link

TPU estimator does not shutdown #412

Open rakshanda22 opened 5 years ago

rakshanda22 commented 5 years ago

When I run the entire code my estimator shuts down automatically, but when running a certain function for prediction the estimator will be initiated but I don't see the shutdown info about the TPU. How can this be forced before returning the value?

ijkilchenko commented 5 years ago

What code are you running specifically?