Add limit_val_batches in JenkinsFile for NMT

1) Add trainer.limit_val_batches in Megatron NMT Training TP=2 2) Remove unused import in ModelPT Signed-off-by: Abhishree <abhishreetm@gmail.com>
NVIDIA · Aug 5, 2023 · c8dab2d · c8dab2d
1 parent 5a60f11
commit c8dab2d
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/Jenkinsfile b/Jenkinsfile
@@ -2545,6 +2545,7 @@ pipeline {
         trainer.accelerator=gpu \
         trainer.log_every_n_steps=1 \
         trainer.val_check_interval=10 \
+        +trainer.limit_val_batches=2 \
         trainer.accumulate_grad_batches=1 \
         trainer.max_steps=10 \
         trainer.precision=16 \

diff --git a/nemo/core/classes/modelPT.py b/nemo/core/classes/modelPT.py
@@ -15,7 +15,6 @@
 
 import copy
 import inspect
-import itertools
 import os
 import uuid
 from abc import abstractmethod