小编o s*_* sy的帖子

运行时错误:混合不同的 tf.distribute.Strategy 对象

你好?我在使用TPU编译模型时遇到了一些问题,部分代码如下:

resolver = tf.contrib.cluster_resolver.TPUClusterResolver(TF_MASTER)

tf.contrib.distribute.initialize_tpu_system(resolver)

strategy = tf.contrib.distribute.TPUStrategy(resolver)

with strategy.scope():

  model = create_model()

  model.compile(optimizer=tf.keras.optimizers.Adadelta(),loss='categorical_crossentropy',metrics='accuracy'])
Run Code Online (Sandbox Code Playgroud)

我得到了 RuntimeError: enter image description here

你能帮助我吗?

google-colaboratory google-cloud-tpu tpu

5
推荐指数
1
解决办法
844
查看次数