Keras - 我的损失可以是模型参数的函数吗？

Question

我注意到在Keras中，优化器采用了

形式的损失函数

def loss(y_true, y_pred):
    //some loss function

如果我的损失函数只是模型参数的函数怎么办？如何在优化器中使用所述损失函数？ 以下是使用Keras优化器的示例：

sgd = optimizers.SGD(lr=0.001, decay=1e-6, 
                  momentum=0.9, nesterov=True)

model.compile(optimizer=sgd,
                  loss=custom_objective,
                  metrics=['accuracy'])

def custom_objective(model.trainable_weights()):
    //some loss function