гугл коллаборация CNN с использованием ТПУ - PullRequest
Новая
       8

гугл коллаборация CNN с использованием ТПУ

0 голосов
/ 29 сентября 2018 
model = larger_model(input_shape)

tpu_grpc_url="grpc://"+os.environ["COLAB_TPU_ADDR"]
tpu_cluster_resolver=tf.contrib.cluster_resolver.TPUClusterResolver(tpu_grpc_url)
strategy=keras_support.TPUDistributionStrategy(tpu_cluster_resolver)

model = tf.contrib.tpu.keras_to_tpu_model(model,strategy=strategy)
model.summary()
tpu_model.fit(x_train,y_train, validation_data=(x_test, y_test), epochs=2, batch_size=8)

здесь модель CNN.Я пытаюсь запустить это на colab, используя tpu, это выдает ошибку как

Checkpointable._track_checkpointable () переданный тип, а не Checkpointable.

в строке 

model = tf.contrib.tpu.keras_to_tpu_model(model,strategy=strategy)

Можете ли вы исправить меня.

...