diff --git a/egs/librispeech/ASR/incremental_transf/.train.py.swp b/egs/librispeech/ASR/incremental_transf/.train.py.swp index 02b0c4faa..941a5a267 100644 Binary files a/egs/librispeech/ASR/incremental_transf/.train.py.swp and b/egs/librispeech/ASR/incremental_transf/.train.py.swp differ diff --git a/egs/librispeech/ASR/incremental_transf/train.py b/egs/librispeech/ASR/incremental_transf/train.py index 02d399655..f024551b5 100755 --- a/egs/librispeech/ASR/incremental_transf/train.py +++ b/egs/librispeech/ASR/incremental_transf/train.py @@ -993,7 +993,10 @@ def run(rank, world_size, args): except: print(f'2: pre-trained model has no parameter named {old_name}.') else: layer_name_splited[0] = 'inter_encoder' + layer_name_splited[1] = 'encoder_layers' layer_name_splited[3] = str(int(layer_name_splited[3])//2) + del layer_name_splited[2] + old_name = '.'.join(layer_name_splited) try: p.data = pre_trained_model2[old_name] except: print(f'3: pre-trained model has no parameter named {old_name}.')