search

wenwenyu / PICK-pytorch

Code for the paper "PICK: Processing Key Information Extraction from Documents using Improved Graph Learning-Convolutional Networks" (ICPR 2020)
https://arxiv.org/abs/2004.07464
MIT License
553 stars 191 forks source link

输出日志里的loss计算 #106

Open shuizhonghaitong opened 2 years ago

shuizhonghaitong commented 2 years ago

在trainer.py中，在backward结束后，为了输出loss会对total_loss做一个all_reduce的操作，但是并未对crf_loss和gl_loss做all_reduce的操作，为什么在后面gl_loss/=size，crf_loss/=size。我理解的难道是因为在对total_loss做all_reduce时，crf_loss和gl_loss也会做相应的改变吗？希望您可以解释一下，谢谢~