Model: "encoder_r1" __________________________________________________________________________________________________ Layer (type) Output Shape Param # Connected to ================================================================================================== input_1 (InputLayer) [(None, 4096, 3)] 0 __________________________________________________________________________________________________ lambda (Lambda) (None, 4096, 2) 0 input_1[0][0] __________________________________________________________________________________________________ tf.__operators__.getitem (Slici (None, 3072, 2) 0 lambda[0][0] __________________________________________________________________________________________________ tf.split (TFOpLambda) [(None, 1024, 2), (N 0 lambda[0][0] __________________________________________________________________________________________________ tf.split_1 (TFOpLambda) [(None, 1024, 2), (N 0 tf.__operators__.getitem[0][0] __________________________________________________________________________________________________ lambda_1 (Lambda) (None, 1024, 2) 0 tf.split[0][0] __________________________________________________________________________________________________ lambda_2 (Lambda) (None, 1024, 2) 0 tf.split_1[0][0] __________________________________________________________________________________________________ lambda_3 (Lambda) (None, 1024, 2) 0 tf.split[0][1] __________________________________________________________________________________________________ lambda_4 (Lambda) (None, 1024, 2) 0 tf.split_1[0][1] __________________________________________________________________________________________________ lambda_5 (Lambda) (None, 1024, 2) 0 tf.split[0][2] __________________________________________________________________________________________________ lambda_6 (Lambda) (None, 1024, 2) 0 tf.split_1[0][2] __________________________________________________________________________________________________ lambda_7 (Lambda) (None, 1024, 2) 0 tf.split[0][3] __________________________________________________________________________________________________ conv1d (Conv1D) (None, 1024, 8) 1032 lambda_1[0][0] __________________________________________________________________________________________________ conv1d_1 (Conv1D) (None, 1024, 8) 1032 lambda_2[0][0] __________________________________________________________________________________________________ conv1d_2 (Conv1D) (None, 1024, 16) 2064 lambda_3[0][0] __________________________________________________________________________________________________ conv1d_3 (Conv1D) (None, 1024, 16) 2064 lambda_4[0][0] __________________________________________________________________________________________________ conv1d_4 (Conv1D) (None, 1024, 16) 2064 lambda_5[0][0] __________________________________________________________________________________________________ conv1d_5 (Conv1D) (None, 1024, 32) 4128 lambda_6[0][0] __________________________________________________________________________________________________ conv1d_6 (Conv1D) (None, 1024, 32) 4128 lambda_7[0][0] __________________________________________________________________________________________________ batch_normalization (BatchNorma (None, 1024, 8) 32 conv1d[0][0] __________________________________________________________________________________________________ batch_normalization_1 (BatchNor (None, 1024, 8) 32 conv1d_1[0][0] __________________________________________________________________________________________________ batch_normalization_2 (BatchNor (None, 1024, 16) 64 conv1d_2[0][0] __________________________________________________________________________________________________ batch_normalization_3 (BatchNor (None, 1024, 16) 64 conv1d_3[0][0] __________________________________________________________________________________________________ batch_normalization_4 (BatchNor (None, 1024, 16) 64 conv1d_4[0][0] __________________________________________________________________________________________________ batch_normalization_5 (BatchNor (None, 1024, 32) 128 conv1d_5[0][0] __________________________________________________________________________________________________ batch_normalization_6 (BatchNor (None, 1024, 32) 128 conv1d_6[0][0] __________________________________________________________________________________________________ leaky_re_lu (LeakyReLU) multiple 0 batch_normalization[0][0] batch_normalization_1[0][0] batch_normalization_2[0][0] batch_normalization_3[0][0] batch_normalization_4[0][0] batch_normalization_5[0][0] batch_normalization_6[0][0] shared_conv_1_batchnorm[0][0] shared_res_3_1st_batchnorm[0][0] shared_res_3_2nd_batchnorm[0][0] add[0][0] shared_res_4_1st_batchnorm[0][0] shared_res_4_2nd_batchnorm[0][0] add_1[0][0] shared_res_6_1st_batchnorm[0][0] shared_res_6_2nd_batchnorm[0][0] add_2[0][0] shared_res_7_1st_batchnorm[0][0] shared_res_7_2nd_batchnorm[0][0] add_3[0][0] shared_res_9_1st_batchnorm[0][0] shared_res_9_2nd_batchnorm[0][0] add_4[0][0] shared_res_10_1st_batchnorm[0][0] shared_res_10_2nd_batchnorm[0][0] add_5[0][0] r1_conv2d_1_batchnorm[0][0] batch_normalization_7[0][0] batch_normalization_8[0][0] add_6[0][0] batch_normalization_10[0][0] batch_normalization_11[0][0] add_7[0][0] batch_normalization_13[0][0] batch_normalization_14[0][0] add_8[0][0] r1_dense_9_batchnorm[0][0] __________________________________________________________________________________________________ concatenate (Concatenate) (None, 1024, 128) 0 leaky_re_lu[0][0] leaky_re_lu[1][0] leaky_re_lu[2][0] leaky_re_lu[3][0] leaky_re_lu[4][0] leaky_re_lu[5][0] leaky_re_lu[6][0] __________________________________________________________________________________________________ shared_conv_1 (Conv1D) (None, 1024, 32) 28704 concatenate[0][0] __________________________________________________________________________________________________ shared_conv_1_batchnorm (BatchN (None, 1024, 32) 128 shared_conv_1[0][0] __________________________________________________________________________________________________ max_pooling1d (MaxPooling1D) (None, 512, 32) 0 leaky_re_lu[7][0] __________________________________________________________________________________________________ shared_res_3_1st (Conv1D) (None, 512, 32) 1056 max_pooling1d[0][0] __________________________________________________________________________________________________ shared_res_3_1st_batchnorm (Bat (None, 512, 32) 128 shared_res_3_1st[0][0] __________________________________________________________________________________________________ shared_res_3_2nd (Conv1D) (None, 512, 32) 7200 leaky_re_lu[8][0] __________________________________________________________________________________________________ shared_res_3_2nd_batchnorm (Bat (None, 512, 32) 128 shared_res_3_2nd[0][0] __________________________________________________________________________________________________ shared_res_3_3rd (Conv1D) (None, 512, 32) 1056 leaky_re_lu[9][0] __________________________________________________________________________________________________ shared_res_3_3rd_batchnorm (Bat (None, 512, 32) 128 shared_res_3_3rd[0][0] __________________________________________________________________________________________________ add (Add) (None, 512, 32) 0 shared_res_3_3rd_batchnorm[0][0] max_pooling1d[0][0] __________________________________________________________________________________________________ shared_res_4_1st (Conv1D) (None, 512, 32) 1056 leaky_re_lu[10][0] __________________________________________________________________________________________________ shared_res_4_1st_batchnorm (Bat (None, 512, 32) 128 shared_res_4_1st[0][0] __________________________________________________________________________________________________ shared_res_4_2nd (Conv1D) (None, 512, 32) 5152 leaky_re_lu[11][0] __________________________________________________________________________________________________ shared_res_4_2nd_batchnorm (Bat (None, 512, 32) 128 shared_res_4_2nd[0][0] __________________________________________________________________________________________________ shared_res_4_3rd (Conv1D) (None, 512, 32) 1056 leaky_re_lu[12][0] __________________________________________________________________________________________________ shared_res_4_3rd_batchnorm (Bat (None, 512, 32) 128 shared_res_4_3rd[0][0] __________________________________________________________________________________________________ add_1 (Add) (None, 512, 32) 0 shared_res_4_3rd_batchnorm[0][0] leaky_re_lu[10][0] __________________________________________________________________________________________________ max_pooling1d_1 (MaxPooling1D) (None, 256, 32) 0 leaky_re_lu[13][0] __________________________________________________________________________________________________ shared_res_6_1st (Conv1D) (None, 256, 32) 1056 max_pooling1d_1[0][0] __________________________________________________________________________________________________ shared_res_6_1st_batchnorm (Bat (None, 256, 32) 128 shared_res_6_1st[0][0] __________________________________________________________________________________________________ shared_res_6_2nd (Conv1D) (None, 256, 32) 5152 leaky_re_lu[14][0] __________________________________________________________________________________________________ shared_res_6_2nd_batchnorm (Bat (None, 256, 32) 128 shared_res_6_2nd[0][0] __________________________________________________________________________________________________ shared_res_6_3rd (Conv1D) (None, 256, 32) 1056 leaky_re_lu[15][0] __________________________________________________________________________________________________ shared_res_6_3rd_batchnorm (Bat (None, 256, 32) 128 shared_res_6_3rd[0][0] __________________________________________________________________________________________________ add_2 (Add) (None, 256, 32) 0 shared_res_6_3rd_batchnorm[0][0] max_pooling1d_1[0][0] __________________________________________________________________________________________________ shared_res_7_1st (Conv1D) (None, 256, 32) 1056 leaky_re_lu[16][0] __________________________________________________________________________________________________ shared_res_7_1st_batchnorm (Bat (None, 256, 32) 128 shared_res_7_1st[0][0] __________________________________________________________________________________________________ shared_res_7_2nd (Conv1D) (None, 256, 32) 5152 leaky_re_lu[17][0] __________________________________________________________________________________________________ shared_res_7_2nd_batchnorm (Bat (None, 256, 32) 128 shared_res_7_2nd[0][0] __________________________________________________________________________________________________ shared_res_7_3rd (Conv1D) (None, 256, 32) 1056 leaky_re_lu[18][0] __________________________________________________________________________________________________ shared_res_7_3rd_batchnorm (Bat (None, 256, 32) 128 shared_res_7_3rd[0][0] __________________________________________________________________________________________________ add_3 (Add) (None, 256, 32) 0 shared_res_7_3rd_batchnorm[0][0] leaky_re_lu[16][0] __________________________________________________________________________________________________ max_pooling1d_2 (MaxPooling1D) (None, 128, 32) 0 leaky_re_lu[19][0] __________________________________________________________________________________________________ shared_res_9_1st (Conv1D) (None, 128, 32) 1056 max_pooling1d_2[0][0] __________________________________________________________________________________________________ shared_res_9_1st_batchnorm (Bat (None, 128, 32) 128 shared_res_9_1st[0][0] __________________________________________________________________________________________________ shared_res_9_2nd (Conv1D) (None, 128, 32) 3104 leaky_re_lu[20][0] __________________________________________________________________________________________________ shared_res_9_2nd_batchnorm (Bat (None, 128, 32) 128 shared_res_9_2nd[0][0] __________________________________________________________________________________________________ shared_res_9_3rd (Conv1D) (None, 128, 32) 1056 leaky_re_lu[21][0] __________________________________________________________________________________________________ shared_res_9_3rd_batchnorm (Bat (None, 128, 32) 128 shared_res_9_3rd[0][0] __________________________________________________________________________________________________ add_4 (Add) (None, 128, 32) 0 shared_res_9_3rd_batchnorm[0][0] max_pooling1d_2[0][0] __________________________________________________________________________________________________ shared_res_10_1st (Conv1D) (None, 128, 32) 1056 leaky_re_lu[22][0] __________________________________________________________________________________________________ shared_res_10_1st_batchnorm (Ba (None, 128, 32) 128 shared_res_10_1st[0][0] __________________________________________________________________________________________________ shared_res_10_2nd (Conv1D) (None, 128, 32) 3104 leaky_re_lu[23][0] __________________________________________________________________________________________________ shared_res_10_2nd_batchnorm (Ba (None, 128, 32) 128 shared_res_10_2nd[0][0] __________________________________________________________________________________________________ shared_res_10_3rd (Conv1D) (None, 128, 32) 1056 leaky_re_lu[24][0] __________________________________________________________________________________________________ shared_res_10_3rd_batchnorm (Ba (None, 128, 32) 128 shared_res_10_3rd[0][0] __________________________________________________________________________________________________ add_5 (Add) (None, 128, 32) 0 shared_res_10_3rd_batchnorm[0][0] leaky_re_lu[22][0] __________________________________________________________________________________________________ max_pooling1d_3 (MaxPooling1D) (None, 64, 32) 0 leaky_re_lu[25][0] __________________________________________________________________________________________________ reshape (Reshape) (None, 64, 32, 1) 0 max_pooling1d_3[0][0] __________________________________________________________________________________________________ r1_conv2d_1 (Conv2D) (None, 64, 32, 32) 832 reshape[0][0] __________________________________________________________________________________________________ r1_conv2d_1_batchnorm (BatchNor (None, 64, 32, 32) 128 r1_conv2d_1[0][0] __________________________________________________________________________________________________ conv2d (Conv2D) (None, 64, 32, 32) 1056 leaky_re_lu[26][0] __________________________________________________________________________________________________ batch_normalization_7 (BatchNor (None, 64, 32, 32) 128 conv2d[0][0] __________________________________________________________________________________________________ conv2d_1 (Conv2D) (None, 64, 32, 32) 25632 leaky_re_lu[27][0] __________________________________________________________________________________________________ batch_normalization_8 (BatchNor (None, 64, 32, 32) 128 conv2d_1[0][0] __________________________________________________________________________________________________ conv2d_2 (Conv2D) (None, 64, 32, 32) 1056 leaky_re_lu[28][0] __________________________________________________________________________________________________ batch_normalization_9 (BatchNor (None, 64, 32, 32) 128 conv2d_2[0][0] __________________________________________________________________________________________________ add_6 (Add) (None, 64, 32, 32) 0 batch_normalization_9[0][0] leaky_re_lu[26][0] __________________________________________________________________________________________________ max_pooling2d (MaxPooling2D) (None, 32, 16, 32) 0 leaky_re_lu[29][0] __________________________________________________________________________________________________ conv2d_3 (Conv2D) (None, 32, 16, 32) 1056 max_pooling2d[0][0] __________________________________________________________________________________________________ batch_normalization_10 (BatchNo (None, 32, 16, 32) 128 conv2d_3[0][0] __________________________________________________________________________________________________ conv2d_4 (Conv2D) (None, 32, 16, 32) 9248 leaky_re_lu[30][0] __________________________________________________________________________________________________ batch_normalization_11 (BatchNo (None, 32, 16, 32) 128 conv2d_4[0][0] __________________________________________________________________________________________________ conv2d_5 (Conv2D) (None, 32, 16, 32) 1056 leaky_re_lu[31][0] __________________________________________________________________________________________________ batch_normalization_12 (BatchNo (None, 32, 16, 32) 128 conv2d_5[0][0] __________________________________________________________________________________________________ add_7 (Add) (None, 32, 16, 32) 0 batch_normalization_12[0][0] max_pooling2d[0][0] __________________________________________________________________________________________________ max_pooling2d_1 (MaxPooling2D) (None, 16, 8, 32) 0 leaky_re_lu[32][0] __________________________________________________________________________________________________ conv2d_6 (Conv2D) (None, 16, 8, 32) 1056 max_pooling2d_1[0][0] __________________________________________________________________________________________________ batch_normalization_13 (BatchNo (None, 16, 8, 32) 128 conv2d_6[0][0] __________________________________________________________________________________________________ conv2d_7 (Conv2D) (None, 16, 8, 32) 9248 leaky_re_lu[33][0] __________________________________________________________________________________________________ batch_normalization_14 (BatchNo (None, 16, 8, 32) 128 conv2d_7[0][0] __________________________________________________________________________________________________ conv2d_8 (Conv2D) (None, 16, 8, 32) 1056 leaky_re_lu[34][0] __________________________________________________________________________________________________ batch_normalization_15 (BatchNo (None, 16, 8, 32) 128 conv2d_8[0][0] __________________________________________________________________________________________________ add_8 (Add) (None, 16, 8, 32) 0 batch_normalization_15[0][0] max_pooling2d_1[0][0] __________________________________________________________________________________________________ max_pooling2d_2 (MaxPooling2D) (None, 8, 4, 32) 0 leaky_re_lu[35][0] __________________________________________________________________________________________________ flatten (Flatten) (None, 1024) 0 max_pooling2d_2[0][0] __________________________________________________________________________________________________ r1_dense_9 (Dense) (None, 1024) 1049600 flatten[0][0] __________________________________________________________________________________________________ r1_dense_9_batchnorm (BatchNorm (None, 1024) 4096 r1_dense_9[0][0] __________________________________________________________________________________________________ r1_mean_dense (Dense) (None, 6) 6150 leaky_re_lu[36][0] __________________________________________________________________________________________________ r1_logvar_dense (Dense) (None, 6) 6150 leaky_re_lu[36][0] __________________________________________________________________________________________________ r1_modes_dense (Dense) (None, 3) 3075 leaky_re_lu[36][0] __________________________________________________________________________________________________ concatenate_1 (Concatenate) (None, 15) 0 r1_mean_dense[0][0] r1_logvar_dense[0][0] r1_modes_dense[0][0] ================================================================================================== Total params: 1,211,343 Trainable params: 1,207,183 Non-trainable params: 4,160 __________________________________________________________________________________________________ Model: "encoder_q" __________________________________________________________________________________________________ Layer (type) Output Shape Param # Connected to ================================================================================================== input_1 (InputLayer) [(None, 4096, 3)] 0 __________________________________________________________________________________________________ lambda (Lambda) (None, 4096, 2) 0 input_1[0][0] __________________________________________________________________________________________________ tf.__operators__.getitem (Slici (None, 3072, 2) 0 lambda[0][0] __________________________________________________________________________________________________ tf.split (TFOpLambda) [(None, 1024, 2), (N 0 lambda[0][0] __________________________________________________________________________________________________ tf.split_1 (TFOpLambda) [(None, 1024, 2), (N 0 tf.__operators__.getitem[0][0] __________________________________________________________________________________________________ lambda_1 (Lambda) (None, 1024, 2) 0 tf.split[0][0] __________________________________________________________________________________________________ lambda_2 (Lambda) (None, 1024, 2) 0 tf.split_1[0][0] __________________________________________________________________________________________________ lambda_3 (Lambda) (None, 1024, 2) 0 tf.split[0][1] __________________________________________________________________________________________________ lambda_4 (Lambda) (None, 1024, 2) 0 tf.split_1[0][1] __________________________________________________________________________________________________ lambda_5 (Lambda) (None, 1024, 2) 0 tf.split[0][2] __________________________________________________________________________________________________ lambda_6 (Lambda) (None, 1024, 2) 0 tf.split_1[0][2] __________________________________________________________________________________________________ lambda_7 (Lambda) (None, 1024, 2) 0 tf.split[0][3] __________________________________________________________________________________________________ conv1d (Conv1D) (None, 1024, 8) 1032 lambda_1[0][0] __________________________________________________________________________________________________ conv1d_1 (Conv1D) (None, 1024, 8) 1032 lambda_2[0][0] __________________________________________________________________________________________________ conv1d_2 (Conv1D) (None, 1024, 16) 2064 lambda_3[0][0] __________________________________________________________________________________________________ conv1d_3 (Conv1D) (None, 1024, 16) 2064 lambda_4[0][0] __________________________________________________________________________________________________ conv1d_4 (Conv1D) (None, 1024, 16) 2064 lambda_5[0][0] __________________________________________________________________________________________________ conv1d_5 (Conv1D) (None, 1024, 32) 4128 lambda_6[0][0] __________________________________________________________________________________________________ conv1d_6 (Conv1D) (None, 1024, 32) 4128 lambda_7[0][0] __________________________________________________________________________________________________ batch_normalization (BatchNorma (None, 1024, 8) 32 conv1d[0][0] __________________________________________________________________________________________________ batch_normalization_1 (BatchNor (None, 1024, 8) 32 conv1d_1[0][0] __________________________________________________________________________________________________ batch_normalization_2 (BatchNor (None, 1024, 16) 64 conv1d_2[0][0] __________________________________________________________________________________________________ batch_normalization_3 (BatchNor (None, 1024, 16) 64 conv1d_3[0][0] __________________________________________________________________________________________________ batch_normalization_4 (BatchNor (None, 1024, 16) 64 conv1d_4[0][0] __________________________________________________________________________________________________ batch_normalization_5 (BatchNor (None, 1024, 32) 128 conv1d_5[0][0] __________________________________________________________________________________________________ batch_normalization_6 (BatchNor (None, 1024, 32) 128 conv1d_6[0][0] __________________________________________________________________________________________________ leaky_re_lu (LeakyReLU) multiple 0 batch_normalization[0][0] batch_normalization_1[0][0] batch_normalization_2[0][0] batch_normalization_3[0][0] batch_normalization_4[0][0] batch_normalization_5[0][0] batch_normalization_6[0][0] shared_conv_1_batchnorm[0][0] shared_res_3_1st_batchnorm[0][0] shared_res_3_2nd_batchnorm[0][0] add[0][0] shared_res_4_1st_batchnorm[0][0] shared_res_4_2nd_batchnorm[0][0] add_1[0][0] shared_res_6_1st_batchnorm[0][0] shared_res_6_2nd_batchnorm[0][0] add_2[0][0] shared_res_7_1st_batchnorm[0][0] shared_res_7_2nd_batchnorm[0][0] add_3[0][0] shared_res_9_1st_batchnorm[0][0] shared_res_9_2nd_batchnorm[0][0] add_4[0][0] shared_res_10_1st_batchnorm[0][0] shared_res_10_2nd_batchnorm[0][0] add_5[0][0] batch_normalization_16[0][0] q_conv2d_1_batchnorm[0][0] batch_normalization_17[0][0] batch_normalization_18[0][0] add_9[0][0] batch_normalization_20[0][0] batch_normalization_21[0][0] add_10[0][0] batch_normalization_23[0][0] batch_normalization_24[0][0] add_11[0][0] q_dense_9_batchnorm[0][0] __________________________________________________________________________________________________ concatenate (Concatenate) (None, 1024, 128) 0 leaky_re_lu[0][0] leaky_re_lu[1][0] leaky_re_lu[2][0] leaky_re_lu[3][0] leaky_re_lu[4][0] leaky_re_lu[5][0] leaky_re_lu[6][0] __________________________________________________________________________________________________ shared_conv_1 (Conv1D) (None, 1024, 32) 28704 concatenate[0][0] __________________________________________________________________________________________________ shared_conv_1_batchnorm (BatchN (None, 1024, 32) 128 shared_conv_1[0][0] __________________________________________________________________________________________________ max_pooling1d (MaxPooling1D) (None, 512, 32) 0 leaky_re_lu[7][0] __________________________________________________________________________________________________ shared_res_3_1st (Conv1D) (None, 512, 32) 1056 max_pooling1d[0][0] __________________________________________________________________________________________________ shared_res_3_1st_batchnorm (Bat (None, 512, 32) 128 shared_res_3_1st[0][0] __________________________________________________________________________________________________ shared_res_3_2nd (Conv1D) (None, 512, 32) 7200 leaky_re_lu[8][0] __________________________________________________________________________________________________ shared_res_3_2nd_batchnorm (Bat (None, 512, 32) 128 shared_res_3_2nd[0][0] __________________________________________________________________________________________________ shared_res_3_3rd (Conv1D) (None, 512, 32) 1056 leaky_re_lu[9][0] __________________________________________________________________________________________________ shared_res_3_3rd_batchnorm (Bat (None, 512, 32) 128 shared_res_3_3rd[0][0] __________________________________________________________________________________________________ add (Add) (None, 512, 32) 0 shared_res_3_3rd_batchnorm[0][0] max_pooling1d[0][0] __________________________________________________________________________________________________ shared_res_4_1st (Conv1D) (None, 512, 32) 1056 leaky_re_lu[10][0] __________________________________________________________________________________________________ shared_res_4_1st_batchnorm (Bat (None, 512, 32) 128 shared_res_4_1st[0][0] __________________________________________________________________________________________________ shared_res_4_2nd (Conv1D) (None, 512, 32) 5152 leaky_re_lu[11][0] __________________________________________________________________________________________________ shared_res_4_2nd_batchnorm (Bat (None, 512, 32) 128 shared_res_4_2nd[0][0] __________________________________________________________________________________________________ shared_res_4_3rd (Conv1D) (None, 512, 32) 1056 leaky_re_lu[12][0] __________________________________________________________________________________________________ shared_res_4_3rd_batchnorm (Bat (None, 512, 32) 128 shared_res_4_3rd[0][0] __________________________________________________________________________________________________ add_1 (Add) (None, 512, 32) 0 shared_res_4_3rd_batchnorm[0][0] leaky_re_lu[10][0] __________________________________________________________________________________________________ max_pooling1d_1 (MaxPooling1D) (None, 256, 32) 0 leaky_re_lu[13][0] __________________________________________________________________________________________________ shared_res_6_1st (Conv1D) (None, 256, 32) 1056 max_pooling1d_1[0][0] __________________________________________________________________________________________________ shared_res_6_1st_batchnorm (Bat (None, 256, 32) 128 shared_res_6_1st[0][0] __________________________________________________________________________________________________ shared_res_6_2nd (Conv1D) (None, 256, 32) 5152 leaky_re_lu[14][0] __________________________________________________________________________________________________ shared_res_6_2nd_batchnorm (Bat (None, 256, 32) 128 shared_res_6_2nd[0][0] __________________________________________________________________________________________________ shared_res_6_3rd (Conv1D) (None, 256, 32) 1056 leaky_re_lu[15][0] __________________________________________________________________________________________________ shared_res_6_3rd_batchnorm (Bat (None, 256, 32) 128 shared_res_6_3rd[0][0] __________________________________________________________________________________________________ add_2 (Add) (None, 256, 32) 0 shared_res_6_3rd_batchnorm[0][0] max_pooling1d_1[0][0] __________________________________________________________________________________________________ shared_res_7_1st (Conv1D) (None, 256, 32) 1056 leaky_re_lu[16][0] __________________________________________________________________________________________________ shared_res_7_1st_batchnorm (Bat (None, 256, 32) 128 shared_res_7_1st[0][0] __________________________________________________________________________________________________ shared_res_7_2nd (Conv1D) (None, 256, 32) 5152 leaky_re_lu[17][0] __________________________________________________________________________________________________ shared_res_7_2nd_batchnorm (Bat (None, 256, 32) 128 shared_res_7_2nd[0][0] __________________________________________________________________________________________________ shared_res_7_3rd (Conv1D) (None, 256, 32) 1056 leaky_re_lu[18][0] __________________________________________________________________________________________________ shared_res_7_3rd_batchnorm (Bat (None, 256, 32) 128 shared_res_7_3rd[0][0] __________________________________________________________________________________________________ add_3 (Add) (None, 256, 32) 0 shared_res_7_3rd_batchnorm[0][0] leaky_re_lu[16][0] __________________________________________________________________________________________________ max_pooling1d_2 (MaxPooling1D) (None, 128, 32) 0 leaky_re_lu[19][0] __________________________________________________________________________________________________ shared_res_9_1st (Conv1D) (None, 128, 32) 1056 max_pooling1d_2[0][0] __________________________________________________________________________________________________ shared_res_9_1st_batchnorm (Bat (None, 128, 32) 128 shared_res_9_1st[0][0] __________________________________________________________________________________________________ shared_res_9_2nd (Conv1D) (None, 128, 32) 3104 leaky_re_lu[20][0] __________________________________________________________________________________________________ shared_res_9_2nd_batchnorm (Bat (None, 128, 32) 128 shared_res_9_2nd[0][0] __________________________________________________________________________________________________ shared_res_9_3rd (Conv1D) (None, 128, 32) 1056 leaky_re_lu[21][0] __________________________________________________________________________________________________ shared_res_9_3rd_batchnorm (Bat (None, 128, 32) 128 shared_res_9_3rd[0][0] __________________________________________________________________________________________________ add_4 (Add) (None, 128, 32) 0 shared_res_9_3rd_batchnorm[0][0] max_pooling1d_2[0][0] __________________________________________________________________________________________________ shared_res_10_1st (Conv1D) (None, 128, 32) 1056 leaky_re_lu[22][0] __________________________________________________________________________________________________ shared_res_10_1st_batchnorm (Ba (None, 128, 32) 128 shared_res_10_1st[0][0] __________________________________________________________________________________________________ shared_res_10_2nd (Conv1D) (None, 128, 32) 3104 leaky_re_lu[23][0] __________________________________________________________________________________________________ shared_res_10_2nd_batchnorm (Ba (None, 128, 32) 128 shared_res_10_2nd[0][0] __________________________________________________________________________________________________ input_2 (InputLayer) [(None, 2)] 0 __________________________________________________________________________________________________ shared_res_10_3rd (Conv1D) (None, 128, 32) 1056 leaky_re_lu[24][0] __________________________________________________________________________________________________ flatten_1 (Flatten) (None, 2) 0 input_2[0][0] __________________________________________________________________________________________________ shared_res_10_3rd_batchnorm (Ba (None, 128, 32) 128 shared_res_10_3rd[0][0] __________________________________________________________________________________________________ q_inx_dense (Dense) (None, 64) 192 flatten_1[0][0] __________________________________________________________________________________________________ add_5 (Add) (None, 128, 32) 0 shared_res_10_3rd_batchnorm[0][0] leaky_re_lu[22][0] __________________________________________________________________________________________________ tf.reshape (TFOpLambda) (None, 64, 1) 0 q_inx_dense[0][0] __________________________________________________________________________________________________ batch_normalization_16 (BatchNo (None, 64, 1) 4 tf.reshape[0][0] __________________________________________________________________________________________________ max_pooling1d_3 (MaxPooling1D) (None, 64, 32) 0 leaky_re_lu[25][0] __________________________________________________________________________________________________ concatenate_2 (Concatenate) (None, 64, 33) 0 max_pooling1d_3[0][0] leaky_re_lu[37][0] __________________________________________________________________________________________________ reshape_1 (Reshape) (None, 64, 33, 1) 0 concatenate_2[0][0] __________________________________________________________________________________________________ q_conv2d_1 (Conv2D) (None, 64, 33, 32) 832 reshape_1[0][0] __________________________________________________________________________________________________ q_conv2d_1_batchnorm (BatchNorm (None, 64, 33, 32) 128 q_conv2d_1[0][0] __________________________________________________________________________________________________ conv2d_9 (Conv2D) (None, 64, 33, 32) 1056 leaky_re_lu[38][0] __________________________________________________________________________________________________ batch_normalization_17 (BatchNo (None, 64, 33, 32) 128 conv2d_9[0][0] __________________________________________________________________________________________________ conv2d_10 (Conv2D) (None, 64, 33, 32) 25632 leaky_re_lu[39][0] __________________________________________________________________________________________________ batch_normalization_18 (BatchNo (None, 64, 33, 32) 128 conv2d_10[0][0] __________________________________________________________________________________________________ conv2d_11 (Conv2D) (None, 64, 33, 32) 1056 leaky_re_lu[40][0] __________________________________________________________________________________________________ batch_normalization_19 (BatchNo (None, 64, 33, 32) 128 conv2d_11[0][0] __________________________________________________________________________________________________ add_9 (Add) (None, 64, 33, 32) 0 batch_normalization_19[0][0] leaky_re_lu[38][0] __________________________________________________________________________________________________ max_pooling2d_3 (MaxPooling2D) (None, 32, 16, 32) 0 leaky_re_lu[41][0] __________________________________________________________________________________________________ conv2d_12 (Conv2D) (None, 32, 16, 32) 1056 max_pooling2d_3[0][0] __________________________________________________________________________________________________ batch_normalization_20 (BatchNo (None, 32, 16, 32) 128 conv2d_12[0][0] __________________________________________________________________________________________________ conv2d_13 (Conv2D) (None, 32, 16, 32) 9248 leaky_re_lu[42][0] __________________________________________________________________________________________________ batch_normalization_21 (BatchNo (None, 32, 16, 32) 128 conv2d_13[0][0] __________________________________________________________________________________________________ conv2d_14 (Conv2D) (None, 32, 16, 32) 1056 leaky_re_lu[43][0] __________________________________________________________________________________________________ batch_normalization_22 (BatchNo (None, 32, 16, 32) 128 conv2d_14[0][0] __________________________________________________________________________________________________ add_10 (Add) (None, 32, 16, 32) 0 batch_normalization_22[0][0] max_pooling2d_3[0][0] __________________________________________________________________________________________________ max_pooling2d_4 (MaxPooling2D) (None, 16, 8, 32) 0 leaky_re_lu[44][0] __________________________________________________________________________________________________ conv2d_15 (Conv2D) (None, 16, 8, 32) 1056 max_pooling2d_4[0][0] __________________________________________________________________________________________________ batch_normalization_23 (BatchNo (None, 16, 8, 32) 128 conv2d_15[0][0] __________________________________________________________________________________________________ conv2d_16 (Conv2D) (None, 16, 8, 32) 9248 leaky_re_lu[45][0] __________________________________________________________________________________________________ batch_normalization_24 (BatchNo (None, 16, 8, 32) 128 conv2d_16[0][0] __________________________________________________________________________________________________ conv2d_17 (Conv2D) (None, 16, 8, 32) 1056 leaky_re_lu[46][0] __________________________________________________________________________________________________ batch_normalization_25 (BatchNo (None, 16, 8, 32) 128 conv2d_17[0][0] __________________________________________________________________________________________________ add_11 (Add) (None, 16, 8, 32) 0 batch_normalization_25[0][0] max_pooling2d_4[0][0] __________________________________________________________________________________________________ max_pooling2d_5 (MaxPooling2D) (None, 8, 4, 32) 0 leaky_re_lu[47][0] __________________________________________________________________________________________________ flatten_2 (Flatten) (None, 1024) 0 max_pooling2d_5[0][0] __________________________________________________________________________________________________ q_dense_9 (Dense) (None, 1024) 1049600 flatten_2[0][0] __________________________________________________________________________________________________ q_dense_9_batchnorm (BatchNorma (None, 1024) 4096 q_dense_9[0][0] __________________________________________________________________________________________________ q_mean_dense (Dense) (None, 2) 2050 leaky_re_lu[48][0] __________________________________________________________________________________________________ q_logvar_dense (Dense) (None, 2) 2050 leaky_re_lu[48][0] __________________________________________________________________________________________________ concatenate_3 (Concatenate) (None, 4) 0 q_mean_dense[0][0] q_logvar_dense[0][0] ================================================================================================== Total params: 1,200,264 Trainable params: 1,196,102 Non-trainable params: 4,162 __________________________________________________________________________________________________ Model: "decoder_r2" __________________________________________________________________________________________________ Layer (type) Output Shape Param # Connected to ================================================================================================== input_1 (InputLayer) [(None, 4096, 3)] 0 __________________________________________________________________________________________________ lambda (Lambda) (None, 4096, 2) 0 input_1[0][0] __________________________________________________________________________________________________ tf.__operators__.getitem (Slici (None, 3072, 2) 0 lambda[0][0] __________________________________________________________________________________________________ tf.split (TFOpLambda) [(None, 1024, 2), (N 0 lambda[0][0] __________________________________________________________________________________________________ tf.split_1 (TFOpLambda) [(None, 1024, 2), (N 0 tf.__operators__.getitem[0][0] __________________________________________________________________________________________________ lambda_1 (Lambda) (None, 1024, 2) 0 tf.split[0][0] __________________________________________________________________________________________________ lambda_2 (Lambda) (None, 1024, 2) 0 tf.split_1[0][0] __________________________________________________________________________________________________ lambda_3 (Lambda) (None, 1024, 2) 0 tf.split[0][1] __________________________________________________________________________________________________ lambda_4 (Lambda) (None, 1024, 2) 0 tf.split_1[0][1] __________________________________________________________________________________________________ lambda_5 (Lambda) (None, 1024, 2) 0 tf.split[0][2] __________________________________________________________________________________________________ lambda_6 (Lambda) (None, 1024, 2) 0 tf.split_1[0][2] __________________________________________________________________________________________________ lambda_7 (Lambda) (None, 1024, 2) 0 tf.split[0][3] __________________________________________________________________________________________________ conv1d (Conv1D) (None, 1024, 8) 1032 lambda_1[0][0] __________________________________________________________________________________________________ conv1d_1 (Conv1D) (None, 1024, 8) 1032 lambda_2[0][0] __________________________________________________________________________________________________ conv1d_2 (Conv1D) (None, 1024, 16) 2064 lambda_3[0][0] __________________________________________________________________________________________________ conv1d_3 (Conv1D) (None, 1024, 16) 2064 lambda_4[0][0] __________________________________________________________________________________________________ conv1d_4 (Conv1D) (None, 1024, 16) 2064 lambda_5[0][0] __________________________________________________________________________________________________ conv1d_5 (Conv1D) (None, 1024, 32) 4128 lambda_6[0][0] __________________________________________________________________________________________________ conv1d_6 (Conv1D) (None, 1024, 32) 4128 lambda_7[0][0] __________________________________________________________________________________________________ batch_normalization (BatchNorma (None, 1024, 8) 32 conv1d[0][0] __________________________________________________________________________________________________ batch_normalization_1 (BatchNor (None, 1024, 8) 32 conv1d_1[0][0] __________________________________________________________________________________________________ batch_normalization_2 (BatchNor (None, 1024, 16) 64 conv1d_2[0][0] __________________________________________________________________________________________________ batch_normalization_3 (BatchNor (None, 1024, 16) 64 conv1d_3[0][0] __________________________________________________________________________________________________ batch_normalization_4 (BatchNor (None, 1024, 16) 64 conv1d_4[0][0] __________________________________________________________________________________________________ batch_normalization_5 (BatchNor (None, 1024, 32) 128 conv1d_5[0][0] __________________________________________________________________________________________________ batch_normalization_6 (BatchNor (None, 1024, 32) 128 conv1d_6[0][0] __________________________________________________________________________________________________ leaky_re_lu (LeakyReLU) multiple 0 batch_normalization[0][0] batch_normalization_1[0][0] batch_normalization_2[0][0] batch_normalization_3[0][0] batch_normalization_4[0][0] batch_normalization_5[0][0] batch_normalization_6[0][0] shared_conv_1_batchnorm[0][0] shared_res_3_1st_batchnorm[0][0] shared_res_3_2nd_batchnorm[0][0] add[0][0] shared_res_4_1st_batchnorm[0][0] shared_res_4_2nd_batchnorm[0][0] add_1[0][0] shared_res_6_1st_batchnorm[0][0] shared_res_6_2nd_batchnorm[0][0] add_2[0][0] shared_res_7_1st_batchnorm[0][0] shared_res_7_2nd_batchnorm[0][0] add_3[0][0] shared_res_9_1st_batchnorm[0][0] shared_res_9_2nd_batchnorm[0][0] add_4[0][0] shared_res_10_1st_batchnorm[0][0] shared_res_10_2nd_batchnorm[0][0] add_5[0][0] batch_normalization_26[0][0] r2_conv2d_1_batchnorm[0][0] batch_normalization_27[0][0] batch_normalization_28[0][0] add_12[0][0] batch_normalization_30[0][0] batch_normalization_31[0][0] add_13[0][0] batch_normalization_33[0][0] batch_normalization_34[0][0] add_14[0][0] r2_dense_9_batchnorm[0][0] __________________________________________________________________________________________________ concatenate (Concatenate) (None, 1024, 128) 0 leaky_re_lu[0][0] leaky_re_lu[1][0] leaky_re_lu[2][0] leaky_re_lu[3][0] leaky_re_lu[4][0] leaky_re_lu[5][0] leaky_re_lu[6][0] __________________________________________________________________________________________________ shared_conv_1 (Conv1D) (None, 1024, 32) 28704 concatenate[0][0] __________________________________________________________________________________________________ shared_conv_1_batchnorm (BatchN (None, 1024, 32) 128 shared_conv_1[0][0] __________________________________________________________________________________________________ max_pooling1d (MaxPooling1D) (None, 512, 32) 0 leaky_re_lu[7][0] __________________________________________________________________________________________________ shared_res_3_1st (Conv1D) (None, 512, 32) 1056 max_pooling1d[0][0] __________________________________________________________________________________________________ shared_res_3_1st_batchnorm (Bat (None, 512, 32) 128 shared_res_3_1st[0][0] __________________________________________________________________________________________________ shared_res_3_2nd (Conv1D) (None, 512, 32) 7200 leaky_re_lu[8][0] __________________________________________________________________________________________________ shared_res_3_2nd_batchnorm (Bat (None, 512, 32) 128 shared_res_3_2nd[0][0] __________________________________________________________________________________________________ shared_res_3_3rd (Conv1D) (None, 512, 32) 1056 leaky_re_lu[9][0] __________________________________________________________________________________________________ shared_res_3_3rd_batchnorm (Bat (None, 512, 32) 128 shared_res_3_3rd[0][0] __________________________________________________________________________________________________ add (Add) (None, 512, 32) 0 shared_res_3_3rd_batchnorm[0][0] max_pooling1d[0][0] __________________________________________________________________________________________________ shared_res_4_1st (Conv1D) (None, 512, 32) 1056 leaky_re_lu[10][0] __________________________________________________________________________________________________ shared_res_4_1st_batchnorm (Bat (None, 512, 32) 128 shared_res_4_1st[0][0] __________________________________________________________________________________________________ shared_res_4_2nd (Conv1D) (None, 512, 32) 5152 leaky_re_lu[11][0] __________________________________________________________________________________________________ shared_res_4_2nd_batchnorm (Bat (None, 512, 32) 128 shared_res_4_2nd[0][0] __________________________________________________________________________________________________ shared_res_4_3rd (Conv1D) (None, 512, 32) 1056 leaky_re_lu[12][0] __________________________________________________________________________________________________ shared_res_4_3rd_batchnorm (Bat (None, 512, 32) 128 shared_res_4_3rd[0][0] __________________________________________________________________________________________________ add_1 (Add) (None, 512, 32) 0 shared_res_4_3rd_batchnorm[0][0] leaky_re_lu[10][0] __________________________________________________________________________________________________ max_pooling1d_1 (MaxPooling1D) (None, 256, 32) 0 leaky_re_lu[13][0] __________________________________________________________________________________________________ shared_res_6_1st (Conv1D) (None, 256, 32) 1056 max_pooling1d_1[0][0] __________________________________________________________________________________________________ shared_res_6_1st_batchnorm (Bat (None, 256, 32) 128 shared_res_6_1st[0][0] __________________________________________________________________________________________________ shared_res_6_2nd (Conv1D) (None, 256, 32) 5152 leaky_re_lu[14][0] __________________________________________________________________________________________________ shared_res_6_2nd_batchnorm (Bat (None, 256, 32) 128 shared_res_6_2nd[0][0] __________________________________________________________________________________________________ shared_res_6_3rd (Conv1D) (None, 256, 32) 1056 leaky_re_lu[15][0] __________________________________________________________________________________________________ shared_res_6_3rd_batchnorm (Bat (None, 256, 32) 128 shared_res_6_3rd[0][0] __________________________________________________________________________________________________ add_2 (Add) (None, 256, 32) 0 shared_res_6_3rd_batchnorm[0][0] max_pooling1d_1[0][0] __________________________________________________________________________________________________ shared_res_7_1st (Conv1D) (None, 256, 32) 1056 leaky_re_lu[16][0] __________________________________________________________________________________________________ shared_res_7_1st_batchnorm (Bat (None, 256, 32) 128 shared_res_7_1st[0][0] __________________________________________________________________________________________________ shared_res_7_2nd (Conv1D) (None, 256, 32) 5152 leaky_re_lu[17][0] __________________________________________________________________________________________________ shared_res_7_2nd_batchnorm (Bat (None, 256, 32) 128 shared_res_7_2nd[0][0] __________________________________________________________________________________________________ shared_res_7_3rd (Conv1D) (None, 256, 32) 1056 leaky_re_lu[18][0] __________________________________________________________________________________________________ shared_res_7_3rd_batchnorm (Bat (None, 256, 32) 128 shared_res_7_3rd[0][0] __________________________________________________________________________________________________ add_3 (Add) (None, 256, 32) 0 shared_res_7_3rd_batchnorm[0][0] leaky_re_lu[16][0] __________________________________________________________________________________________________ max_pooling1d_2 (MaxPooling1D) (None, 128, 32) 0 leaky_re_lu[19][0] __________________________________________________________________________________________________ shared_res_9_1st (Conv1D) (None, 128, 32) 1056 max_pooling1d_2[0][0] __________________________________________________________________________________________________ shared_res_9_1st_batchnorm (Bat (None, 128, 32) 128 shared_res_9_1st[0][0] __________________________________________________________________________________________________ shared_res_9_2nd (Conv1D) (None, 128, 32) 3104 leaky_re_lu[20][0] __________________________________________________________________________________________________ shared_res_9_2nd_batchnorm (Bat (None, 128, 32) 128 shared_res_9_2nd[0][0] __________________________________________________________________________________________________ shared_res_9_3rd (Conv1D) (None, 128, 32) 1056 leaky_re_lu[21][0] __________________________________________________________________________________________________ shared_res_9_3rd_batchnorm (Bat (None, 128, 32) 128 shared_res_9_3rd[0][0] __________________________________________________________________________________________________ add_4 (Add) (None, 128, 32) 0 shared_res_9_3rd_batchnorm[0][0] max_pooling1d_2[0][0] __________________________________________________________________________________________________ shared_res_10_1st (Conv1D) (None, 128, 32) 1056 leaky_re_lu[22][0] __________________________________________________________________________________________________ shared_res_10_1st_batchnorm (Ba (None, 128, 32) 128 shared_res_10_1st[0][0] __________________________________________________________________________________________________ shared_res_10_2nd (Conv1D) (None, 128, 32) 3104 leaky_re_lu[23][0] __________________________________________________________________________________________________ shared_res_10_2nd_batchnorm (Ba (None, 128, 32) 128 shared_res_10_2nd[0][0] __________________________________________________________________________________________________ input_3 (InputLayer) [(None, 2)] 0 __________________________________________________________________________________________________ shared_res_10_3rd (Conv1D) (None, 128, 32) 1056 leaky_re_lu[24][0] __________________________________________________________________________________________________ flatten_3 (Flatten) (None, 2) 0 input_3[0][0] __________________________________________________________________________________________________ shared_res_10_3rd_batchnorm (Ba (None, 128, 32) 128 shared_res_10_3rd[0][0] __________________________________________________________________________________________________ r2_inz_dense (Dense) (None, 64) 192 flatten_3[0][0] __________________________________________________________________________________________________ add_5 (Add) (None, 128, 32) 0 shared_res_10_3rd_batchnorm[0][0] leaky_re_lu[22][0] __________________________________________________________________________________________________ tf.reshape_1 (TFOpLambda) (None, 64, 1) 0 r2_inz_dense[0][0] __________________________________________________________________________________________________ batch_normalization_26 (BatchNo (None, 64, 1) 4 tf.reshape_1[0][0] __________________________________________________________________________________________________ max_pooling1d_3 (MaxPooling1D) (None, 64, 32) 0 leaky_re_lu[25][0] __________________________________________________________________________________________________ concatenate_4 (Concatenate) (None, 64, 33) 0 max_pooling1d_3[0][0] leaky_re_lu[49][0] __________________________________________________________________________________________________ reshape_2 (Reshape) (None, 64, 33, 1) 0 concatenate_4[0][0] __________________________________________________________________________________________________ r2_conv2d_1 (Conv2D) (None, 64, 33, 32) 832 reshape_2[0][0] __________________________________________________________________________________________________ r2_conv2d_1_batchnorm (BatchNor (None, 64, 33, 32) 128 r2_conv2d_1[0][0] __________________________________________________________________________________________________ conv2d_18 (Conv2D) (None, 64, 33, 32) 1056 leaky_re_lu[50][0] __________________________________________________________________________________________________ batch_normalization_27 (BatchNo (None, 64, 33, 32) 128 conv2d_18[0][0] __________________________________________________________________________________________________ conv2d_19 (Conv2D) (None, 64, 33, 32) 25632 leaky_re_lu[51][0] __________________________________________________________________________________________________ batch_normalization_28 (BatchNo (None, 64, 33, 32) 128 conv2d_19[0][0] __________________________________________________________________________________________________ conv2d_20 (Conv2D) (None, 64, 33, 32) 1056 leaky_re_lu[52][0] __________________________________________________________________________________________________ batch_normalization_29 (BatchNo (None, 64, 33, 32) 128 conv2d_20[0][0] __________________________________________________________________________________________________ add_12 (Add) (None, 64, 33, 32) 0 batch_normalization_29[0][0] leaky_re_lu[50][0] __________________________________________________________________________________________________ max_pooling2d_6 (MaxPooling2D) (None, 32, 16, 32) 0 leaky_re_lu[53][0] __________________________________________________________________________________________________ conv2d_21 (Conv2D) (None, 32, 16, 32) 1056 max_pooling2d_6[0][0] __________________________________________________________________________________________________ batch_normalization_30 (BatchNo (None, 32, 16, 32) 128 conv2d_21[0][0] __________________________________________________________________________________________________ conv2d_22 (Conv2D) (None, 32, 16, 32) 9248 leaky_re_lu[54][0] __________________________________________________________________________________________________ batch_normalization_31 (BatchNo (None, 32, 16, 32) 128 conv2d_22[0][0] __________________________________________________________________________________________________ conv2d_23 (Conv2D) (None, 32, 16, 32) 1056 leaky_re_lu[55][0] __________________________________________________________________________________________________ batch_normalization_32 (BatchNo (None, 32, 16, 32) 128 conv2d_23[0][0] __________________________________________________________________________________________________ add_13 (Add) (None, 32, 16, 32) 0 batch_normalization_32[0][0] max_pooling2d_6[0][0] __________________________________________________________________________________________________ max_pooling2d_7 (MaxPooling2D) (None, 16, 8, 32) 0 leaky_re_lu[56][0] __________________________________________________________________________________________________ conv2d_24 (Conv2D) (None, 16, 8, 32) 1056 max_pooling2d_7[0][0] __________________________________________________________________________________________________ batch_normalization_33 (BatchNo (None, 16, 8, 32) 128 conv2d_24[0][0] __________________________________________________________________________________________________ conv2d_25 (Conv2D) (None, 16, 8, 32) 9248 leaky_re_lu[57][0] __________________________________________________________________________________________________ batch_normalization_34 (BatchNo (None, 16, 8, 32) 128 conv2d_25[0][0] __________________________________________________________________________________________________ conv2d_26 (Conv2D) (None, 16, 8, 32) 1056 leaky_re_lu[58][0] __________________________________________________________________________________________________ batch_normalization_35 (BatchNo (None, 16, 8, 32) 128 conv2d_26[0][0] __________________________________________________________________________________________________ add_14 (Add) (None, 16, 8, 32) 0 batch_normalization_35[0][0] max_pooling2d_7[0][0] __________________________________________________________________________________________________ max_pooling2d_8 (MaxPooling2D) (None, 8, 4, 32) 0 leaky_re_lu[59][0] __________________________________________________________________________________________________ flatten_4 (Flatten) (None, 1024) 0 max_pooling2d_8[0][0] __________________________________________________________________________________________________ r2_dense_9 (Dense) (None, 1024) 1049600 flatten_4[0][0] __________________________________________________________________________________________________ r2_dense_9_batchnorm (BatchNorm (None, 1024) 4096 r2_dense_9[0][0] __________________________________________________________________________________________________ JointVonMisesFisher_mean (Dense (None, 3) 3075 leaky_re_lu[60][0] __________________________________________________________________________________________________ JointVonMisesFisher_logvar (Den (None, 1) 1025 leaky_re_lu[60][0] __________________________________________________________________________________________________ concatenate_5 (Concatenate) (None, 4) 0 JointVonMisesFisher_mean[0][0] JointVonMisesFisher_logvar[0][0] ================================================================================================== Total params: 1,200,264 Trainable params: 1,196,102 Non-trainable params: 4,162 __________________________________________________________________________________________________