from local

egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py

@@ -206,6 +206,13 @@ def get_parser():
         default=12354,
         help="Master port to use for DDP training.",
     )
+    
+    parser.add_argument(
+        "--wandb",
+        type=str2bool,
+        default=False,
+        help="Should various information be logged in wandb.",
+    )
 
     parser.add_argument(
         "--tensorboard",