diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.prompt_tuning.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.prompt_tuning.py.swp index c3a5a85bc..73dfd6ef8 100644 Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.prompt_tuning.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.prompt_tuning.py.swp differ diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/prompt_tuning.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/prompt_tuning.py index c2386263f..adc77f8f9 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/prompt_tuning.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/prompt_tuning.py @@ -1535,7 +1535,7 @@ def run_adapter(rank, world_size, args, wb=None): params = get_params() params.update(vars(args)) - #fix_random_seed(params.seed) + fix_random_seed(params.seed) if world_size > 1: setup_dist(rank, world_size, params.master_port) @@ -1670,7 +1670,7 @@ def run_adapter(rank, world_size, args, wb=None): for epoch in range(params.start_epoch, params.num_epochs + 1): logging.info(f"update num : {params.batch_idx_train}") scheduler.step_epoch(epoch - 1) - #fix_random_seed(params.seed + epoch - 1) + fix_random_seed(params.seed + epoch - 1) train_dl.sampler.set_epoch(epoch - 1) if tb_writer is not None: @@ -1693,7 +1693,6 @@ def run_adapter(rank, world_size, args, wb=None): rank=rank, wb=wb, ) - exit() if params.print_diagnostics: diagnostic.print_diagnostics()