torch-rechub icon indicating copy to clipboard operation
torch-rechub copied to clipboard

Published 20 hours ago • verified publisher icon datawhalechina

loss监控

Open ZoeyChen-lab opened this issue 1 year ago • 0 comments

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

ZoeyChen-lab avatar Jun 27 '23 07:06 ZoeyChen-lab