Блог

Главная - Вопросы по программированию - как исправить ошибку NotImplementedError

как исправить ошибку NotImplementedError

#python #tensorflow

Вопрос:

У меня есть следующий код для запуска vgg

 checkpoint = tf.keras.callbacks.ModelCheckpoint(checkpoint_filepath 'MAssCalcVGG-noAug.h5', 
        monitor='val_loss', mode='auto', verbose=1, 
        save_best_only=True, save_freq='epoch'
)
#add custom fully-connected network on top of the already-trained base network 
model = models.Sequential()
model.add(conv_base)
model.add(layers.Flatten())
model.add(layers.Dense(256, activation="relu"))
model.add(layers.Dense(1, activation="sigmoid"))

#freeze convolutional base 
conv_base.trainable = False

model.compile(loss="binary_crossentropy",
            optimizer=optimizers.Adam(lr=1e-3), # lr = 0.0001
            metrics=METRICS) 

#train fully-connected added part 
history = model.fit(train_generat.flow(train_dataset_split,
                                       train_labels_split,
                                       batch_size=BATCH_SIZE,
                                       shuffle=False),
                    steps_per_epoch=len(train_dataset_split) // BATCH_SIZE, 
                    epochs=100,
                    validation_data=valid_generat.flow(valid_dataset_split,
                                       valid_labels_split,
                                       batch_size=BATCH_SIZE,
                                       shuffle=False),
                    validation_steps=len(valid_labels_split) // BATCH_SIZE,
                    callbacks=[es, checkpoint, GarbageCollectorCallback()])
#model.save(save(os.path.join(checkpoint_filepath, 'MAssCalcVGG-noAug.h5'))))

model.summary()

но я понимаю это:

 NotImplementedError: Layer ModuleWrapper has arguments in `__init__` and therefore must override `get_config`

как я могу это исправить?
код отлично работал до того, как, вероятно, я намеренно внес изменения, чтобы вызвать это.

Комментарии:

1. Не могли бы вы поделиться своей conv_base реализацией слоя?

2. conv_base = VGG16(веса=’imagenet’, include_top=False, input_shape=(150, 150, 3)) conv_base.summary()

3. Вы получаете эту ошибку, когда хотите сохранить свою модель (строка с комментарием в вашем вопросе) или без сохранения модели? В какой строке вы получаете эту ошибку?

4. не сохранив его. однако теперь моя проблема решена. Я действительно не знаю как :D. может быть, это была незначительная ошибка, которую я сделал с контрольной точкой и перезвонил. потому что я переписываю их, используя свой старый код, и теперь он ухудшается

5. Вы смешиваете импорт между keras и tf.keras? Добавьте все импортные данные в свой вопрос.

Метки: