diff --git a/egs/librispeech/ASR/incremental_transf/.identity_train.py.swp b/egs/librispeech/ASR/incremental_transf/.identity_train.py.swp index b727a9154..2f9ecb4c5 100644 Binary files a/egs/librispeech/ASR/incremental_transf/.identity_train.py.swp and b/egs/librispeech/ASR/incremental_transf/.identity_train.py.swp differ diff --git a/egs/librispeech/ASR/incremental_transf/identity_train.py b/egs/librispeech/ASR/incremental_transf/identity_train.py index 25b3cf20f..4ebc01a48 100755 --- a/egs/librispeech/ASR/incremental_transf/identity_train.py +++ b/egs/librispeech/ASR/incremental_transf/identity_train.py @@ -972,11 +972,14 @@ def run(rank, world_size, args): for n, p in model.named_parameters(): if 'layer' not in n: try: p.data = pre_trained_model[n] - except: print(f'pre-trained model has no parameterd named {n}.') + except: print(f'pre-trained model has no parameter named {n}.') else: layer_name_splited = n.split('.') if int(layer_name_splited[3]) % 2 == 0: layer_name_splited[3] = str(int(layer_name_splited[3])//2) + old_name = '.'.join(layer_name_splited) + try: p.data = pre_trained_model[old_name] + except: print(f'pre-trained model has no parameter named {n}.') print('.'.join(layer_name_splited)) exit()