- 博客(2)
- 收藏
- 关注
原创 tensorflow2 加载h5 Unknown loss function
一共有两种解法: 1. 在load_model方法中,将compile设置为False model = keras.models.load_model('PCB.h5', compile=False) 2. 用custom_objects属性重新定义损失函数,custom_loss是函数名字 t=tf.keras.models.load_model(f'{model_path}',custom_objects={'define_loss':custom_loss}) ...
2022-02-23 10:40:37 690
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人