diff --git a/egs/librispeech/ASR/incremental_transf/.identity_train.py.swp b/egs/librispeech/ASR/incremental_transf/.identity_train.py.swp index 155700a10..aa591accb 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 a06ae258a..020f01ac5 100755 --- a/egs/librispeech/ASR/incremental_transf/identity_train.py +++ b/egs/librispeech/ASR/incremental_transf/identity_train.py @@ -974,8 +974,11 @@ def run(rank, world_size, args): try: p.data = pre_trained_model[n] except: print(f'pre-trained model has no parameterd named {n}.') else: - - print(n[23:25]) + layer_name_splited = n.split('.') + if int(layer_name_splited[3]) % 2 == 0: + layer_name_splited[3] = str(int(layer_name_splited[3])//2) + print(''.join(layer_name_splited)) + exit() num_param = sum([p.numel() for p in model.parameters()])