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
e902ed6efa
commit
8be9c5779f
Binary file not shown.
@ -966,7 +966,7 @@ def run(rank, world_size, args):
|
|||||||
logging.info("About to create model")
|
logging.info("About to create model")
|
||||||
model = get_transducer_model(params)
|
model = get_transducer_model(params)
|
||||||
|
|
||||||
pre_trained_model = torch.load('/workspace/icefall/egs/librispeech/ASR/incremental_trans/conformer_12layers.pt')
|
pre_trained_model = torch.load('/workspace/icefall/egs/librispeech/ASR/incremental_transf/conformer_12layers.pt')
|
||||||
pre_trained_model = pre_trained_model['model']
|
pre_trained_model = pre_trained_model['model']
|
||||||
model.load_state_dict(pre_trained_model, strict=True)
|
model.load_state_dict(pre_trained_model, strict=True)
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user