models.py 文件源码

def compile_masked(model, lr=0.0005, num_classes=2):
        beta_1 = 0.9
        beta_2 = 0.999
        epsilon = 10 ** (-8)
        optimizer = Adam(lr=lr, beta_1=beta_1, beta_2=beta_2, epsilon=epsilon, clipnorm=1.)

        loss = [lambda y_true, y_pred: y_pred]

        model.compile(
            optimizer=optimizer,
            loss=loss,

        )
        return model