search

median-research-group / LibMTL

A PyTorch Library for Multi-Task Learning
MIT License
1.98k stars 185 forks source link

训练效果会受到epoch大小的影响吗，我只有一个任务同时使用6个loss，想看一下效果 #76

Closed entuo closed 7 months ago

entuo commented 7 months ago

还有一个问题，这个train_loss_buffer到底是如何更新的。self.model.train_loss_buffer[:, epoch] = self.meter.loss_item在train中

Baijiong-Lin commented 7 months ago

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

不同数据集的情况可能不一样，建议你自己尝试一下

还有一个问题，这个train_loss_buffer到底是如何更新的。self.model.train_loss_buffer[:, epoch] = self.meter.loss_item在train中

每个epoch的平均loss

entuo commented 7 months ago

谢谢您的答复