`watch` with epoch - Githubissues

UETAILab / uetai

Custom ML tracking experiment and debugging tools.

https://uetailab.github.io/uetai/

MIT License

15 stars 1 forks source link

`watch` with epoch #17

Open nmd2k opened 2 years ago

nmd2k commented 2 years ago

Describe the bug The default of watch is logging by step. It should be called in every epoch end to monitor the change of gradient of the model.

To Reproduce Run a dummy training with uetai.SummaryWriter

logger = uetai.SummaryWriter(project='uetai')
...
logger.watch(model, criterion)

Screenshots

Expected behavior Gradient:

Parameter:

Desktop (please complete the following information):

nmd2k commented 2 years ago

Nonsense charts

[ ] Epoch chart

[ ] Trainer global step chart