from local

This commit is contained in:
dohe0342 2023-01-09 22:48:13 +09:00
parent 44d0f6e0fb
commit 9af1c4234f
2 changed files with 1 additions and 1 deletions

View File

@ -977,7 +977,7 @@ def run(rank, world_size, args):
try: try:
path = '/workspace/icefall/egs/librispeech/ASR/incremental_transf/conformer_12layers.pt' path = '/workspace/icefall/egs/librispeech/ASR/incremental_transf/conformer_12layers.pt'
load_checkpoint(transducer_model) load_checkpoint(path, transducer_model)
except: except:
path = '/home/work/workspace/icefall/egs/librispeech/ASR/incremental_transf/conformer_12layers.pt' path = '/home/work/workspace/icefall/egs/librispeech/ASR/incremental_transf/conformer_12layers.pt'
pre_trained_model = pre_trained_model['model'] pre_trained_model = pre_trained_model['model']