diff --git a/funasr/build_utils/build_trainer.py b/funasr/build_utils/build_trainer.py index 060b57f46..71fc6df7d 100644 --- a/funasr/build_utils/build_trainer.py +++ b/funasr/build_utils/build_trainer.py @@ -75,14 +75,14 @@ class TrainerOptions: grad_clip: float grad_clip_type: float log_interval: Optional[int] - no_forward_run: bool + # no_forward_run: bool use_tensorboard: bool - use_wandb: bool + # use_wandb: bool output_dir: Union[Path, str] max_epoch: int max_update: int seed: int - sharded_ddp: bool + # sharded_ddp: bool patience: Optional[int] keep_nbest_models: Union[int, List[int]] nbest_averaging_interval: int @@ -90,7 +90,7 @@ class TrainerOptions: best_model_criterion: Sequence[Sequence[str]] val_scheduler_criterion: Sequence[str] unused_parameters: bool - wandb_model_log_interval: int + # wandb_model_log_interval: int use_pai: bool oss_bucket: Union[oss2.Bucket, None]