mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
915e8e399c
- Added CHiME-4 dataset integration in asr_datamodule.py - Added Hugging Face upload script - Added RIR augmentation - Added Self-Distillation Training
19 lines
458 B
Python
Executable File
19 lines
458 B
Python
Executable File
# 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)
|