如何使用自定义的损失函数训练模型？ about modelscope HOT 2 CLOSED

你好，loss是由模型的forward方法返回的，也就是说由模型负责计算，您可以对模型进行继承或定制，不需要定制trainer。
具体的方法：

1. 查看被训练模型中，loss function的定义，继承该模型并覆盖掉loss function
2. 新的模型上加上注解，模型类型自定义一个
3. 在trainer的cfg_modify_fn中，将cfg的model.type字段改为新的模型类型

代码看起来就像这样：
`
@MODELS.register_model('old-task-name', 'new-model-name')
class NewModel(ToBeTrainedModel):
def init(...):
# may be change the loss function here
...
def forward(...):
# may be change the loss function here
...

def cfg_modify_fn(cfg):
cfg.model.type = 'new-model-name'
# other changes here
...

trainer = build_trainer('trainer', model='some-model-dir-or-id', cfg_modify_fn=cfg_modify_fn, **other_kwargs)
trainer.train()
`

from modelscope.

close as "resolved"

from modelscope.