Merge pull request #2 from zzasdf/k2ssl-util

checkpoint convert script

egs/librispeech/SSL/local/convert_checkpoint_from_fairseq.py

@@ -0,0 +1,18 @@
+# simple script to convert a fairseq checkpoint into pytorch parameter state dict
+from argparse import ArgumentParser
+from collections import OrderedDict
+
+import torch
+
+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)