Bug in epoch calculation

Open dineshkh opened this issue 3 years ago • 0 comments

At line no. 1483 in the file codegen_sources/model/src/trainer.py. the code is self.n_sentences += params.batch_size I think it should be self.n_sentences += len1.size(0) https://github.com/facebookresearch/CodeGen/blob/6e93aca63e7bc77287c9965a5080456326651237/codegen_sources/model/src/trainer.py#L1483

With above bug notion of one epoch becomes wrong because of check at following line.

https://github.com/facebookresearch/CodeGen/blob/6e93aca63e7bc77287c9965a5080456326651237/codegen_sources/model/train.py#L742

Jul 20 '22 16:07 dineshkh