如何更新DWA中的train_loss_buffer - Githubissues

median-research-group / LibMTL

A PyTorch Library for Multi-Task Learning

MIT License

1.94k stars 181 forks source link

如何更新DWA中的train_loss_buffer #74

Closed entuo closed 5 months ago

Baijiong-Lin commented 5 months ago

https://github.com/median-research-group/LibMTL/blob/d2d84fa242597cdbb1d128c98a1f761f133c54ec/LibMTL/trainer.py#L237

entuo commented 5 months ago

训练的效果会受到epoch的大小影响吗