from local

2025-12-11 06:55:27 +00:00 · 2023-01-09 22:42:14 +09:00 · 2023-01-09 22:42:14 +09:00 · 2c0d1bb8ff
commit 2c0d1bb8ff
parent c6fd716758
3 changed files with 1 additions and 2 deletions
--- a/egs/librispeech/ASR/incremental_transf/.identity_train.py.swp
+++ b/egs/librispeech/ASR/incremental_transf/.identity_train.py.swp
--- a/egs/librispeech/ASR/incremental_transf/identity_train.py
+++ b/egs/librispeech/ASR/incremental_transf/identity_train.py
@ -979,10 +979,9 @@ def run(rank, world_size, args):
    except: pre_trained_model = torch.load('/home/work/workspace/icefall/egs/librispeech/ASR/incremental_transf/conformer_12layers.pt')
    pre_trained_model = pre_trained_model['model']
    transducer_model.load_state_dict(pre_trained_model, strict=True)
+    transducer_model.to(device)
    model = get_interformer_model(transducer_model.encoder, params)

-    del pre_trained_model
-
    for n, p in model.named_parameters():
        if 'pt_encoder' in n:
            p.requires_grad = False
--- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.checkpoint.py.swp
+++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.checkpoint.py.swp