update

嘉渊

2023-04-25 622c025e10459375e91544885f7a138d41989892

update

 funasr/bin/train.py

@@ -260,6 +260,12 @@
             "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 @@
    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