diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.full_ft.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.full_ft.py.swp
index 35bbd66af..fdd988f77 100644
Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.full_ft.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.full_ft.py.swp differ
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.prompt_tuning.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.prompt_tuning.py.swp
index 54cb6ce2e..ca26fd0cd 100644
Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.prompt_tuning.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.prompt_tuning.py.swp differ
diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/full_ft.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/full_ft.py
index a688b12d3..1267c1da2 100755
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/full_ft.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/full_ft.py
@@ -333,6 +333,13 @@ def get_parser():
         default=30,
         help="Number of epochs to train.",
     )
+    
+    parser.add_argument(
+        "--num-updates",
+        type=int,
+        default=5000,
+        help="Number of epochs to train.",
+    )
 
     parser.add_argument(
         "--start-epoch",