search

datawhalechina / torch-rechub

A Lighting Pytorch Framework for Recommendation Models, Easy-to-use and Easy-to-extend.
MIT License
374 stars 69 forks source link

loss监控 #54

Open ZoeyChen-lab opened 1 year ago

ZoeyChen-lab commented 1 year ago

我在使用召回模型的时候想要对模型训练的过程添加监控，请问在直接使用torch_rehub库时如何能够获取到torch_rechub/trainers/match_trainer/train_one_epoch中的loss？是否能添加一个返回变量？