From 952abee88c59a8bcaa07ef20f3dedb2aabca3885 Mon Sep 17 00:00:00 2001
From: zzasdf <15218404468@163.com>
Date: Tue, 19 Mar 2024 17:29:34 +0800
Subject: [PATCH] add checkpoint convert script

---
 .../local/convert_checkpoint_from_fairseq.py    | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)
 create mode 100644 egs/librispeech/SSL/local/convert_checkpoint_from_fairseq.py

diff --git a/egs/librispeech/SSL/local/convert_checkpoint_from_fairseq.py b/egs/librispeech/SSL/local/convert_checkpoint_from_fairseq.py
new file mode 100644
index 000000000..f06adb9e5
--- /dev/null
+++ b/egs/librispeech/SSL/local/convert_checkpoint_from_fairseq.py
@@ -0,0 +1,17 @@
+# simple script to convert a fairseq checkpoint into pytorch parameter state dict
+import torch
+from collections import OrderedDict
+from argparse import ArgumentParser
+
+parser = ArgumentParser()
+parser.add_argument("--src")
+parser.add_argument("--tgt")
+
+args = parser.parse_args()
+src = args.src
+tgt = args.tgt
+
+old_checkpoint = torch.load(src)
+new_checkpoint = OrderedDict()
+new_checkpoint['model'] = old_checkpoint['model']
+torch.save(new_checkpoint, tgt)