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
a05d12c637
commit
a731bf4bab
Binary file not shown.
@ -966,9 +966,9 @@ 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('/home/work/workspace/icefall/egs/librispeech/ASR/incremental_trans/conformer_12layers.pt')
|
pre_trained_model = torch.load('/workspace/icefall/egs/librispeech/ASR/incremental_trans/conformer_12layers.pt')
|
||||||
pre_trained_model = pre_trained_model['model']
|
pre_trained_model = pre_trained_model['model']
|
||||||
model.load_state_dict(pre_trained_model)
|
model.load_state_dict(pre_trained_model, strict=True)
|
||||||
|
|
||||||
'''
|
'''
|
||||||
for n, p in model.named_parameters():
|
for n, p in model.named_parameters():
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user