update

funasr/bin/train.py

@@ -260,6 +260,12 @@ def get_parser():
              "training phase. If None is given, it is decided according the number "
              "of training samples automatically .",
     )
+    parser.add_argument(
+        "--use_tensorboard",
+        type=str2bool,
+        default=True,
+        help="Enable tensorboard logging",
+    )
 
     # pretrained model related
     parser.add_argument(

funasr/build_utils/build_trainer.py

@@ -76,7 +76,7 @@ class TrainerOptions:
     grad_clip_type: float
     log_interval: Optional[int]
     # no_forward_run: bool
-    # use_tensorboard: bool
+    use_tensorboard: bool
     # use_wandb: bool
     output_dir: Union[Path, str]
     max_epoch: int