shithub: opus

Info  •  Files  •  Log  •  Branches

added missing dead-zone to encode_rdovae.py

--- a/dnn/training_tf2/encode_rdovae.py

+++ b/dnn/training_tf2/encode_rdovae.py

@@ -48,6 +48,8 @@

 import importlib

 rdovae = importlib.import_module(args.model)

+from rdovae import apply_dead_zone

+

 import sys

 import numpy as np

 from tensorflow.keras.optimizers import Adam

@@ -105,10 +107,15 @@

 features.astype('float32').tofile(args.output + "-input.f32")

 #quant_out.astype('float32').tofile(args.output + "-enc_dec.f32")

+nbits=80

+dead_zone = tf.math.softplus(quant_embed_dec[:, :, nbits : 2 * nbits])

+symbols = apply_dead_zone([bits, dead_zone]).numpy()

 np.round(bits).astype('int16').tofile(args.output + "-bits.s16")

 quant_embed_dec.astype('float32').tofile(args.output + "-quant.f32")

 gru_state_dec = gru_state_dec[:,-1,:]

 dec_out = decoder([bits[:,1::2,:], quant_embed_dec[:,1::2,:], gru_state_dec])

+

+print(dec_out.shape)

 dec_out.numpy().astype('float32').tofile(args.output + "-dec_out.f32")

-- 

⑨