Keras load model 导入自定义函数
Keras 是一个强大的深度学习框架,我习惯用 ModelCheckpoint 的方式将模型保存为 hdf5 格式,然后再用 load_model 函数加载模型。今天跑实验的时候却报了一个 error,如下:
ValueError: Unknown loss function: dice_coef_loss
原来 load_model 加载的是默认的 loss function, 而 dice_coef_loss 是我自己定义的 loss function,那么怎么调用自定义的函数呢?答案是加一个 cu
原创
2021-03-12 22:13:54 ·
2357 阅读 ·
0 评论