mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
14700684ab
commit
51404f1c69
Binary file not shown.
@ -976,7 +976,7 @@ def run(rank, world_size, args):
|
||||
transducer_model = get_transducer_model(params)
|
||||
|
||||
try: pre_trained_model = torch.load('/workspace/icefall/egs/librispeech/ASR/incremental_transf/conformer_12layers.pt')
|
||||
except: pre_trained_model = torch.load('/workspace/icefall/egs/librispeech/ASR/incremental_transf/conformer_12layers.pt')
|
||||
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)
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user