Bug in epoch calculation
dineshkh opened this issue · 0 comments
dineshkh commented
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)
CodeGen/codegen_sources/model/src/trainer.py
Line 1483 in 6e93aca
With above bug notion of one epoch becomes wrong because of check at following line.
CodeGen/codegen_sources/model/train.py
Line 742 in 6e93aca