mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-09 10:02:22 +00:00
87843e9382
* Add k2SSL * fix flake8 * fix for black * fix for black * fix for black * Update ssl_datamodule.py * Fix bugs in HubertDataset * update comments * add librilight * add checkpoint convert script * format --------- Co-authored-by: yifanyeung <yifanyeung@yifanyeung.local> Co-authored-by: zzasdf <15218404468@163.com>
19 lines
458 B
Python
19 lines
458 B
Python
# 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)
|