Open dineshkh opened 2 years ago
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
self.n_sentences += params.batch_size
self.n_sentences += len1.size(0)
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
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 beself.n_sentences += len1.size(0)
https://github.com/facebookresearch/CodeGen/blob/6e93aca63e7bc77287c9965a5080456326651237/codegen_sources/model/src/trainer.py#L1483With 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