标签: python-2.7 google-cloud-tpu
我正在尝试使用更多的TPU内核进行培训。我的原始代码使用keras创建模型,并使用keras_to_tpu_model转换模型以在TPU内核上运行。但是我需要降低速度并提高性能。看来我无法通过任何模块更改主机号或使用的内核。我该怎么办?
keras
keras_to_tpu_model