search

gathierry / FastFlow

Apache License 2.0
130 stars 22 forks source link

where can I get the val loss? #20

Open FrankFang0813 opened 1 year ago

FrankFang0813 commented 1 year ago

Hi, first, thank you for your code!!!

I want to use val loss to choose the best ckpt, I think "loss_meter.avg" is training loss. So where can I get the val loss? thank you!!

gathierry commented 1 year ago

You can get the loss value from ret here with ret["loss"] https://github.com/gathierry/FastFlow/blob/37c99ae33b8db0e5cb534fd0d197b0bc44817a03/main.py#L96

Then build another loss_meter in eval_once function to aggregate the losses.