mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2023-04-27 21:12:37 +09:00
parent ae1b318493
commit e5e0363fb2
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.prompt_tuning.py.swp
View File

Binary file not shown.

5
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/prompt_tuning.py
View File

@ -1535,7 +1535,7 @@ def run_adapter(rank, world_size, args, wb=None):
params = get_params() params = get_params()
params.update(vars(args)) params.update(vars(args))
#fix_random_seed(params.seed) fix_random_seed(params.seed)
if world_size > 1: if world_size > 1:
setup_dist(rank, world_size, params.master_port) 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): for epoch in range(params.start_epoch, params.num_epochs + 1):
logging.info(f"update num : {params.batch_idx_train}") logging.info(f"update num : {params.batch_idx_train}")
scheduler.step_epoch(epoch - 1) 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) train_dl.sampler.set_epoch(epoch - 1)
if tb_writer is not None: if tb_writer is not None:
@ -1693,7 +1693,6 @@ def run_adapter(rank, world_size, args, wb=None):
rank=rank, rank=rank,
wb=wb, wb=wb,
) )
exit()
if params.print_diagnostics: if params.print_diagnostics:
diagnostic.print_diagnostics() diagnostic.print_diagnostics()