search

victoresque / pytorch-template

PyTorch deep learning projects made easy.
MIT License
4.7k stars 1.08k forks source link

Readme #61

Closed berniel closed 4 years ago

berniel commented 4 years ago

in readme: Additional logging

If you have additional information to be logged, in _train_epoch() of your trainer class, merge them with log as shown below before returning:

additional_log = {"gradient_norm": g, "sensitivity": s} log = log.update(additional_log) return log

should it be _log.update(additionallog) instead of _log = log.update(additionallog)?

SunQpark commented 4 years ago

Yes it should... Thank you for reporting this.