diff --git a/egs/librispeech/ASR/incremental_transf/.train.py.swp b/egs/librispeech/ASR/incremental_transf/.train.py.swp index 6a7d650f8..5dc3a5e57 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 f740e1167..19e81e32b 100755 --- a/egs/librispeech/ASR/incremental_transf/train.py +++ b/egs/librispeech/ASR/incremental_transf/train.py @@ -974,6 +974,10 @@ def run(rank, world_size, args): pre_trained2 = torch.load(path2) pre_trained_model2 = pre_trained2["model"] + for n, p in pre_trained_model2.items(): + if rank == 0: + print(n) + for n, p in model.named_parameters(): if 'layer' not in n: try: p.data = pre_trained_model1[n]