diff --git a/.idea/mmsr.iml b/.idea/mmsr.iml index 00d8c774..b63a5458 100644 --- a/.idea/mmsr.iml +++ b/.idea/mmsr.iml @@ -13,7 +13,7 @@ - \ No newline at end of file diff --git a/codes/train.py b/codes/train.py index 8fea3b20..9f7349ce 100644 --- a/codes/train.py +++ b/codes/train.py @@ -286,7 +286,7 @@ class Trainer: if __name__ == '__main__': parser = argparse.ArgumentParser() - parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../options/train_mel_encoder_pred_codes.yml') + parser.add_argument('-opt', type=str, help='Path to option YAML file.', default='../options/train_gpt_asr_mass_hf2.yml') parser.add_argument('--launcher', choices=['none', 'pytorch'], default='none', help='job launcher') parser.add_argument('--local_rank', type=int, default=0) args = parser.parse_args()