加在model时:
model = load_model('model/multi_task/try.h5', custom_objects={'loss_max': loss_max})
自定义的loss function:
def loss_max(y_true, y_pred):
from keras import backend as K
return K.max(K.abs(y_pred - y_true), axis=-1)
加在model时:
model = load_model('model/multi_task/try.h5', custom_objects={'loss_max': loss_max})
自定义的loss function:
def loss_max(y_true, y_pred):
from keras import backend as K
return K.max(K.abs(y_pred - y_true), axis=-1)