search

facebookresearch / ClassyVision

An end-to-end PyTorch framework for image and video classification
https://classyvision.ai
MIT License
1.59k stars 278 forks source link

how to print 'sub loss' #751

Closed fangwudi closed 3 years ago

fangwudi commented 3 years ago

How can I log values of each sub loss in SumArbitraryLoss or MultiOutputSumLoss just like the way in keras . For exsample: tain log, total loss: x, sub loss 1: x1, sub loss 2: x2 .

mannatsingh commented 3 years ago

This isn't directly supported. But you should be able to create a hook that derives from LossLrMeterLoggingHook and override this section - https://github.com/facebookresearch/ClassyVision/blob/master/classy_vision/hooks/loss_lr_meter_logging_hook.py#L72-L77

fangwudi commented 3 years ago

https://github.com/facebookresearch/ClassyVision/blob/master/classy_vision/hooks/loss_lr_meter_logging_hook.py#L72-L77

Ok, thanks.