TypeError: Tensor n'est pas partageable si l'égalité Tensor est activée. Utilisez plutôt tensor.experimental_ref () comme clé
J'essayais d'appliquer l'apprentissage par transfert à InceptionV3. Voici mon code:
inception_model = InceptionV3(weights='imagenet',include_top=False)
output_inception = inception_model.output
output_globalavgpooling = GlobalAveragePooling2D()(output_inception)
output_dense = Dense(1024,activation='relu')(output_globalavgpooling)
predictions = Dense(1,activation='sigmoid')(output_dense)
final_model = Model(inception_model.input,output=predictions)
final_model.compile()
inception_model.summary()
Lorsque j'exécute ce code, j'obtiens l'erreur suivante sur la ligne final_model = Model(inception_model.input,output=predictions):
TypeError: Tensor is unhashable if Tensor equality is enabled. Instead, use tensor.experimental_ref() as the key.
Que devrais-je faire?
Avez-vous essayé cela?
final_model = tf.compat.v1.keras.Model(inception_model.input,output=predictions)