ApexEDM / PEBG

53 stars 15 forks source link

关于train_loss计算 #7

Open coder-zc opened 1 year ago

coder-zc commented 1 year ago

请问在pebg_dkt.py中第164行 train_loss /= batch_loss,为什么每一个epoch的train_loss要除以batch_loss,不是应该除以batch_size吗?

另外我运行了您的代码,发现tarin_loss基本维持不变,而且前几个epoch的auc值已经接近best_auc了,我不知道继续训练的意义是什么,希望你可以抽空解答一下。

amazing-2020 commented 7 months ago

估计已经毕业好几年了