small fix for NMT build job

sillsdev · Sep 3, 2024 · 1b848ec · 1b848ec
1 parent 368120f
commit 1b848ec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/machine/jobs/nmt_engine_build_job.py b/machine/jobs/nmt_engine_build_job.py
@@ -74,7 +74,7 @@ def _train_model(
         ) as model_trainer:
             model_trainer.train(progress=phase_progress, check_canceled=check_canceled)
             model_trainer.save()
-            train_corpus_size = model_trainer.stats.train_corpus_size
+            train_corpus_size = parallel_corpus.count()
         return train_corpus_size, float("nan")
 
     def _batch_inference(