diff --git a/dnn/training_tf2/lpcnet.py b/dnn/training_tf2/lpcnet.py index cbd73d83..f450ba5a 100644 --- a/dnn/training_tf2/lpcnet.py +++ b/dnn/training_tf2/lpcnet.py @@ -268,6 +268,7 @@ def new_lpcnet_model(rnn_units1=384, rnn_units2=16, nb_used_features=20, batch_s past_errors = error_calc([Input_extractor([pcm,0]),tensor_preds]) embed = diff_Embed(name='embed_sig',initializer = PCMInit()) cpcm = Concatenate()([tf_l2u(Input_extractor([pcm,0])),tf_l2u(tensor_preds),past_errors]) + cpcm = GaussianNoise(.3)(cpcm) cpcm = Reshape((-1, embed_size*3))(embed(cpcm)) cpcm_decoder = Concatenate()([Input_extractor([dpcm,0]),Input_extractor([dpcm,1]),Input_extractor([dpcm,2])]) cpcm_decoder = Reshape((-1, embed_size*3))(embed(cpcm_decoder))