from local

2025-12-11 06:55:27 +00:00 · 2023-04-11 16:03:39 +09:00 · 2023-04-11 16:03:39 +09:00 · ecb581d7db
commit ecb581d7db
parent 7cf6aaa4ef
2 changed files with 2 additions and 0 deletions
--- 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
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/prompt_tuning.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/prompt_tuning.py
@ -1616,6 +1616,8 @@ def run_adapter(rank, world_size, args, wb=None):
    
    librispeech = LibriSpeechAsrDataModule(args)

+    prompt = prompt.to('cuda')
+    
    '''
    if params.hpo:
        train_cuts = librispeech.train_clean_10_cuts(option=params.gender)