added LPCNet torch implementation

Signed-off-by: Jan Buethe <jbuethe@amazon.de>
This commit is contained in:
Jan Buethe 2023-09-05 12:29:38 +02:00
parent 90a171c1c2
commit 35ee397e06
No known key found for this signature in database
GPG key ID: 9E32027A35B36314
38 changed files with 3200 additions and 0 deletions

View file

@ -0,0 +1,25 @@
""" script for updating checkpoints with new setup entries
Use this script to update older outputs with newly introduced
parameters. (Saves us the trouble of backward compatibility)
"""
import argparse
import torch
parser = argparse.ArgumentParser()
parser.add_argument('checkpoint_file', type=str, help='checkpoint to be updated')
parser.add_argument('--model', type=str, help='model update', default=None)
args = parser.parse_args()
checkpoint = torch.load(args.checkpoint_file, map_location='cpu')
# update model entry
if type(args.model) != type(None):
checkpoint['setup']['lpcnet']['model'] = args.model
torch.save(checkpoint, args.checkpoint_file)