shithub: opus

Info  •  Files  •  Log  •  Branches

fixed use of quant_levels argument in fec_encoder

--- a/dnn/training_tf2/fec_encoder.py

+++ b/dnn/training_tf2/fec_encoder.py

@@ -73,7 +73,7 @@

     args = parser.parse_args()

-model, encoder, decoder, qembedding = new_rdovae_model(nb_used_features=20, nb_bits=80, batch_size=1, cond_size=args.cond_size)

+model, encoder, decoder, qembedding = new_rdovae_model(nb_used_features=20, nb_bits=80, batch_size=1, nb_quant=args.quant_levels, cond_size=args.cond_size)

 model.load_weights(args.weights)

 lpc_order = 16

-- 

⑨