Python 无法在Google Colab中打开文件(文件读取失败:time=Sun Aug 23 12:19:51 2020…)
我正在谷歌Colab中训练一个用于图像分割的神经网络 培训代码为:Python 无法在Google Colab中打开文件(文件读取失败:time=Sun Aug 23 12:19:51 2020…),python,neural-network,google-colaboratory,Python,Neural Network,Google Colaboratory,我正在谷歌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, p
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')
作为:
在此方面的任何帮助都将不胜感激。您正在尝试从目录加载模型。请注意,命令.load_weights()需要.h5文件扩展名,当然不是目录 我想你在找这个:
model=tf.saved\u model.load(路径)