from local

egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_adapter.py

@@ -1464,17 +1464,18 @@ def run(rank, world_size, args, wb=None):
         if params.print_diagnostics:
             diagnostic.print_diagnostics()
             break
-
+        
-        save_checkpoint(
+        if epoch % 10 == 0:
-            params=params,
+            save_checkpoint(
-            model=model,
+                params=params,
-            model_avg=model_avg,
+                model=model,
-            optimizer=optimizer,
+                model_avg=model_avg,
-            scheduler=scheduler,
+                optimizer=optimizer,
-            sampler=train_dl.sampler,
+                scheduler=scheduler,
-            scaler=scaler,
+                sampler=train_dl.sampler,
-            rank=rank,
+                scaler=scaler,
-        )
+                rank=rank,
+            )
 
     logging.info("Done!")