关于Colab上Keras模型转TPU模型的问题

使用TPU加速训练,将Keras模型转TPU模型时报错,如图图片说明

关键代码如下
引用库:

%tensorflow_version 1.x
import json
import os

import numpy as np
import tensorflow as tf
from  tensorflow.python.keras.applications import resnet
from  tensorflow.python.keras import callbacks
from  tensorflow.python.keras.preprocessing.image import ImageDataGenerator
import matplotlib.pyplot as plt

转换TPU模型代码如下

 # This address identifies the TPU we'll use when configuring TensorFlow.
        TPU_WORKER = 'grpc://' + os.environ['COLAB_TPU_ADDR']
        tf.logging.set_verbosity(tf.logging.INFO)

        self.model = tf.contrib.tpu.keras_to_tpu_model(self.model, strategy=tf.contrib.tpu.TPUDistributionStrategy(tf.contrib.cluster_resolver.TPUClusterResolver(TPU_WORKER)))
        self.model = resnet50.ResNet50(weights=None, input_shape=dataset.input_shape, classes=num_classes)

1个回答

Csdn user default icon
上传中...
上传图片
插入图片
抄袭、复制答案,以达到刷声望分或其他目的的行为,在CSDN问答是严格禁止的,一经发现立刻封号。是时候展现真正的技术了!
立即提问
相关内容推荐