eden-emu/opus
15
0
Fork 0
mirror of https://github.com/xiph/opus.git synced 2025-05-27 21:59:12 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

added quant-levels argument to dump_rdovae and rdovae_exchange

Browse source
This commit is contained in:
jbuethe 2022-11-08 11:35:38 +00:00
parent 9859d68bb0
commit ecb5cbcf30
2 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
dnn/training_tf2/dump_rdovae.py
View file

@ -39,6 +39,7 @@ parser = argparse.ArgumentParser()
parser.add_argument('weights', metavar="<weight file>", type=str, help='model weight file in hdf5 format')
parser.add_argument('--cond-size', type=int, help="conditioning size (default: 256)", default=256)
parser.add_argument('--latent-dim', type=int, help="dimension of latent space (default: 80)", default=80)
parser.add_argument('--quant-levels', type=int, help="number of quantization steps (default: 16)", default=16)
args = parser.parse_args()
@ -114,7 +115,7 @@ extern const opus_uint16 dred_p0_q15[{levels * N}];
if __name__ == "__main__":
model, encoder, decoder, qembedding = new_rdovae_model(20, args.latent_dim, cond_size=args.cond_size)
model, encoder, decoder, qembedding = new_rdovae_model(20, args.latent_dim, cond_size=args.cond_size, nb_quant=args.quant_levels)
model.load_weights(args.weights)

3
dnn/training_tf2/rdovae_exchange.py
View file

@ -40,6 +40,7 @@ parser.add_argument('weights', metavar="<weight file>", type=str, help='model we
parser.add_argument('output', metavar="<output folder>", type=str, help='output exchange folder')
parser.add_argument('--cond-size', type=int, help="conditioning size (default: 256)", default=256)
parser.add_argument('--latent-dim', type=int, help="dimension of latent space (default: 80)", default=80)
parser.add_argument('--quant-levels', type=int, help="number of quantization steps (default: 16)", default=16)
args = parser.parse_args()
@ -78,7 +79,7 @@ exchange_name = {
if __name__ == "__main__":
model, encoder, decoder, qembedding = new_rdovae_model(20, args.latent_dim, cond_size=args.cond_size)
model, encoder, decoder, qembedding = new_rdovae_model(20, args.latent_dim, cond_size=args.cond_size, nb_quant=args.quant_levels)
model.load_weights(args.weights)
os.makedirs(args.output, exist_ok=True)