diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.bias_compare.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.bias_compare.py.swp index d14719651..9bacff0af 100644 Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.bias_compare.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.bias_compare.py.swp differ diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/bias_compare.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/bias_compare.py new file mode 100644 index 000000000..878c8ec51 --- /dev/null +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/bias_compare.py @@ -0,0 +1,10 @@ +import torch + +base_model = torch.load('./d2v-base-T.pt') +bias_model = torch.load('./bitfit_2414_v2/checkpoint-100.pt') + +base_model, bias_model = base_model['model'], bias_model['model'] + +for key in base_model.keys(): + if 'bias' in key: + print(key, base_model[key], bias_model[key])