How can I train mnasnet using CPU or GPU on GCP

tensorflow / tpu

Reference models and tools for Cloud TPUs.

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

Apache License 2.0

5.21k stars 1.77k forks source link

How can I train mnasnet using CPU or GPU on GCP #445

Closed mylovekj closed 5 years ago

mylovekj commented 5 years ago

I have used GCP VM to run mnasnet using CPU.

I set the parameter "--use_tpu=false" but result message is "RuntimeError: You need to specify a TPU Name if you are running in the Google Cloud environment."

Is there any other way to run mnasnet on GCP not use TPU?

saberkun commented 5 years ago

Thank you for reporting. This is clear a bug. Will update with a fix soon.

saberkun commented 5 years ago

I think setting tpu_cluster_resolver to None will fix your issue. I had commit: https://github.com/tensorflow/tpu/commit/83b7420c8fe261eb55359caa7fab8f9c02b81706

let us know if it is fixed.

mylovekj commented 5 years ago

Thank you. It work.

mylovekj commented 5 years ago

But I have one more question. Can I specify tensorflow worker parameter?