作者热门文章
- html - 出于某种原因,IE8 对我的 Sass 文件中继承的 html5 CSS 不友好?
- JMeter 在响应断言中使用 span 标签的问题
- html - 在 :hover and :active? 上具有不同效果的 CSS 动画
- html - 相对于居中的 html 内容固定的 CSS 重复背景?
我正在尝试使用 TPU 选项在 Google Colab 上运行一个简单的 MNIST 分类器。使用 Keras 创建模型后,我尝试通过以下方式将其转换为 TPU:
import tensorflow as tf
import os
tpu_model = tf.contrib.tpu.keras_to_tpu_model(
model,
strategy=tf.contrib.tpu.TPUDistributionStrategy(
tf.contrib.cluster_resolver.TPUClusterResolver(tpu='grpc://' + os.environ['COLAB_TPU_ADDR'])
)
)
tpu_model.compile(
optimizer='rmsprop',
loss='categorical_crossentropy',
metrics=['accuracy']
)
print(model.summary())
我得到的错误是:
---------------------------------------------------------------------------
KeyError Traceback (most recent call last)
<ipython-input-5-63c528142aab> in <module>()
5 model,
6 strategy=tf.contrib.tpu.TPUDistributionStrategy(
----> 7 tf.contrib.cluster_resolver.TPUClusterResolver(tpu='grpc://' + os.environ['COLAB_TPU_ADDR'])
8 )
9 )
/usr/lib/python3.6/os.py in __getitem__(self, key)
667 except KeyError:
668 # raise KeyError with the original key value
--> 669 raise KeyError(key) from None
670 return self.decodevalue(value)
671
KeyError: 'COLAB_TPU_ADDR'
看起来我需要更改 TPU 地址,但一直在谷歌搜索,但还没有找到任何东西。感谢您的帮助,谢谢!
最佳答案
关于deep-learning - Google Colab KeyError : 'COLAB_TPU_ADDR' ,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/53142474/
我正在尝试使用 TPU 选项在 Google Colab 上运行一个简单的 MNIST 分类器。使用 Keras 创建模型后,我尝试通过以下方式将其转换为 TPU: import tensorflow
我是一名优秀的程序员,十分优秀!