From 41d5eeb2dde066e46e5e7605e3576ceefe3cc6a0 Mon Sep 17 00:00:00 2001 From: yanghaha0908 Date: Thu, 23 May 2024 20:02:06 +0800 Subject: [PATCH] pr --- examples/asr_librispeech/finetune_asr.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/examples/asr_librispeech/finetune_asr.py b/examples/asr_librispeech/finetune_asr.py index dbb3a13e..7f659e73 100644 --- a/examples/asr_librispeech/finetune_asr.py +++ b/examples/asr_librispeech/finetune_asr.py @@ -15,6 +15,9 @@ class RunConfig: fsdp_config: FSDPConfig = field(default_factory=FSDPConfig) debug: bool = field(default=False, metadata={"help": "Use pdb when true"}) metric: str = field(default="acc", metadata={"help": "The metric for evaluation"}) + ckpt_path: str = field( + default="output/model.pt", metadata={"help": "The path to projector checkpoint"} + ) @hydra.main(config_name=None, version_base=None) def main_hydra(cfg: DictConfig):