python - 无法在 Google Colab 中打开文件（文件读取失败：时间 = Sun Aug 23 12:19:51 2020...）

Question

我正在 Google Colab 中训练用于图像分割的神经网络。

训练代码如下：

 print('Training')
 batch_size = 8
 nb_epoch = 5
 mcp_save = ModelCheckpoint('weight_isic18', save_best_only=True, monitor='val_loss', mode='min')
 reduce_lr_loss = ReduceLROnPlateau(monitor='val_loss', factor=0.1, patience=7, verbose=1, epsilon=1e-4, mode='min')
 history = model.fit(tr_data,tr_mask,
          batch_size=batch_size,
          epochs=nb_epoch,
          shuffle=True,
          verbose=1,
          validation_data=(val_data, val_mask), callbacks=[mcp_save, reduce_lr_loss] )
 print('Trained model saved')
 with open('hist_isic18', 'wb') as file_pi:
    pickle.dump(history.history, file_pi)

训练过程顺利，权重保存在这里：

加载权重时出现错误：

model.load_weights('weight_isic18')

作为：

在这方面的任何帮助将不胜感激。

标签： pythonneural-networkgoogle-colaboratory

Answer 1

您正在尝试从目录加载模型。请注意，命令 .load_weights() 需要 .h5 文件扩展名，当然不是目录。

我想你正在寻找这个：
model = tf.saved_model.load(path)