Model: "encoder_r1" __________________________________________________________________________________________________ Layer (type) Output Shape Param # Connected to ================================================================================================== input_1 (InputLayer) [(None, 4096, 3)] 0 __________________________________________________________________________________________________ shared_conv_0 (Conv1D) (None, 4096, 64) 12544 input_1[0][0] __________________________________________________________________________________________________ shared_conv_0_batchnorm (BatchN (None, 4096, 64) 256 shared_conv_0[0][0] __________________________________________________________________________________________________ leaky_re_lu (LeakyReLU) multiple 0 shared_conv_0_batchnorm[0][0] shared_res_1_1st_batchnorm[0][0] shared_res_1_2nd_batchnorm[0][0] add[0][0] shared_res_3_1st_batchnorm[0][0] shared_res_3_2nd_batchnorm[0][0] add_1[0][0] shared_conv_5_batchnorm[0][0] shared_res_6_1st_batchnorm[0][0] shared_res_6_2nd_batchnorm[0][0] add_2[0][0] shared_res_8_1st_batchnorm[0][0] shared_res_8_2nd_batchnorm[0][0] add_3[0][0] shared_conv_10_batchnorm[0][0] shared_res_11_1st_batchnorm[0][0] shared_res_11_2nd_batchnorm[0][0] add_4[0][0] shared_res_13_1st_batchnorm[0][0] shared_res_13_2nd_batchnorm[0][0] add_5[0][0] shared_conv_15_batchnorm[0][0] shared_res_16_1st_batchnorm[0][0] shared_res_16_2nd_batchnorm[0][0] add_6[0][0] shared_res_18_1st_batchnorm[0][0] shared_res_18_2nd_batchnorm[0][0] add_7[0][0] shared_conv_20_batchnorm[0][0] shared_res_21_1st_batchnorm[0][0] shared_res_21_2nd_batchnorm[0][0] add_8[0][0] shared_res_23_1st_batchnorm[0][0] shared_res_23_2nd_batchnorm[0][0] add_9[0][0] r1_dense_1_batchnorm[0][0] r1_dense_2_batchnorm[0][0] r1_dense_3_batchnorm[0][0] r1_dense_4_batchnorm[0][0] __________________________________________________________________________________________________ shared_res_1_1st (Conv1D) (None, 4096, 64) 4160 leaky_re_lu[0][0] __________________________________________________________________________________________________ shared_res_1_1st_batchnorm (Bat (None, 4096, 64) 256 shared_res_1_1st[0][0] __________________________________________________________________________________________________ shared_res_1_2nd (Conv1D) (None, 4096, 64) 266304 leaky_re_lu[1][0] __________________________________________________________________________________________________ shared_res_1_2nd_batchnorm (Bat (None, 4096, 64) 256 shared_res_1_2nd[0][0] __________________________________________________________________________________________________ shared_res_1_3rd (Conv1D) (None, 4096, 64) 4160 leaky_re_lu[2][0] __________________________________________________________________________________________________ shared_res_1_3rd_batchnorm (Bat (None, 4096, 64) 256 shared_res_1_3rd[0][0] __________________________________________________________________________________________________ add (Add) (None, 4096, 64) 0 shared_res_1_3rd_batchnorm[0][0] leaky_re_lu[0][0] __________________________________________________________________________________________________ max_pooling1d (MaxPooling1D) (None, 2047, 64) 0 leaky_re_lu[3][0] __________________________________________________________________________________________________ shared_res_3_1st (Conv1D) (None, 2047, 64) 4160 max_pooling1d[0][0] __________________________________________________________________________________________________ shared_res_3_1st_batchnorm (Bat (None, 2047, 64) 256 shared_res_3_1st[0][0] __________________________________________________________________________________________________ shared_res_3_2nd (Conv1D) (None, 2047, 64) 266304 leaky_re_lu[4][0] __________________________________________________________________________________________________ shared_res_3_2nd_batchnorm (Bat (None, 2047, 64) 256 shared_res_3_2nd[0][0] __________________________________________________________________________________________________ shared_res_3_3rd (Conv1D) (None, 2047, 64) 4160 leaky_re_lu[5][0] __________________________________________________________________________________________________ shared_res_3_3rd_batchnorm (Bat (None, 2047, 64) 256 shared_res_3_3rd[0][0] __________________________________________________________________________________________________ add_1 (Add) (None, 2047, 64) 0 shared_res_3_3rd_batchnorm[0][0] max_pooling1d[0][0] __________________________________________________________________________________________________ max_pooling1d_1 (MaxPooling1D) (None, 1023, 64) 0 leaky_re_lu[6][0] __________________________________________________________________________________________________ shared_conv_5 (Conv1D) (None, 1023, 128) 139392 max_pooling1d_1[0][0] __________________________________________________________________________________________________ shared_conv_5_batchnorm (BatchN (None, 1023, 128) 512 shared_conv_5[0][0] __________________________________________________________________________________________________ shared_res_6_1st (Conv1D) (None, 1023, 128) 16512 leaky_re_lu[7][0] __________________________________________________________________________________________________ shared_res_6_1st_batchnorm (Bat (None, 1023, 128) 512 shared_res_6_1st[0][0] __________________________________________________________________________________________________ shared_res_6_2nd (Conv1D) (None, 1023, 128) 278656 leaky_re_lu[8][0] __________________________________________________________________________________________________ shared_res_6_2nd_batchnorm (Bat (None, 1023, 128) 512 shared_res_6_2nd[0][0] __________________________________________________________________________________________________ shared_res_6_3rd (Conv1D) (None, 1023, 128) 16512 leaky_re_lu[9][0] __________________________________________________________________________________________________ shared_res_6_3rd_batchnorm (Bat (None, 1023, 128) 512 shared_res_6_3rd[0][0] __________________________________________________________________________________________________ add_2 (Add) (None, 1023, 128) 0 shared_res_6_3rd_batchnorm[0][0] leaky_re_lu[7][0] __________________________________________________________________________________________________ max_pooling1d_2 (MaxPooling1D) (None, 511, 128) 0 leaky_re_lu[10][0] __________________________________________________________________________________________________ shared_res_8_1st (Conv1D) (None, 511, 128) 16512 max_pooling1d_2[0][0] __________________________________________________________________________________________________ shared_res_8_1st_batchnorm (Bat (None, 511, 128) 512 shared_res_8_1st[0][0] __________________________________________________________________________________________________ shared_res_8_2nd (Conv1D) (None, 511, 128) 278656 leaky_re_lu[11][0] __________________________________________________________________________________________________ shared_res_8_2nd_batchnorm (Bat (None, 511, 128) 512 shared_res_8_2nd[0][0] __________________________________________________________________________________________________ shared_res_8_3rd (Conv1D) (None, 511, 128) 16512 leaky_re_lu[12][0] __________________________________________________________________________________________________ shared_res_8_3rd_batchnorm (Bat (None, 511, 128) 512 shared_res_8_3rd[0][0] __________________________________________________________________________________________________ add_3 (Add) (None, 511, 128) 0 shared_res_8_3rd_batchnorm[0][0] max_pooling1d_2[0][0] __________________________________________________________________________________________________ max_pooling1d_3 (MaxPooling1D) (None, 255, 128) 0 leaky_re_lu[13][0] __________________________________________________________________________________________________ shared_conv_10 (Conv1D) (None, 255, 256) 164096 max_pooling1d_3[0][0] __________________________________________________________________________________________________ shared_conv_10_batchnorm (Batch (None, 255, 256) 1024 shared_conv_10[0][0] __________________________________________________________________________________________________ shared_res_11_1st (Conv1D) (None, 255, 256) 65792 leaky_re_lu[14][0] __________________________________________________________________________________________________ shared_res_11_1st_batchnorm (Ba (None, 255, 256) 1024 shared_res_11_1st[0][0] __________________________________________________________________________________________________ shared_res_11_2nd (Conv1D) (None, 255, 256) 327936 leaky_re_lu[15][0] __________________________________________________________________________________________________ shared_res_11_2nd_batchnorm (Ba (None, 255, 256) 1024 shared_res_11_2nd[0][0] __________________________________________________________________________________________________ shared_res_11_3rd (Conv1D) (None, 255, 256) 65792 leaky_re_lu[16][0] __________________________________________________________________________________________________ shared_res_11_3rd_batchnorm (Ba (None, 255, 256) 1024 shared_res_11_3rd[0][0] __________________________________________________________________________________________________ add_4 (Add) (None, 255, 256) 0 shared_res_11_3rd_batchnorm[0][0] leaky_re_lu[14][0] __________________________________________________________________________________________________ max_pooling1d_4 (MaxPooling1D) (None, 127, 256) 0 leaky_re_lu[17][0] __________________________________________________________________________________________________ shared_res_13_1st (Conv1D) (None, 127, 256) 65792 max_pooling1d_4[0][0] __________________________________________________________________________________________________ shared_res_13_1st_batchnorm (Ba (None, 127, 256) 1024 shared_res_13_1st[0][0] __________________________________________________________________________________________________ shared_res_13_2nd (Conv1D) (None, 127, 256) 327936 leaky_re_lu[18][0] __________________________________________________________________________________________________ shared_res_13_2nd_batchnorm (Ba (None, 127, 256) 1024 shared_res_13_2nd[0][0] __________________________________________________________________________________________________ shared_res_13_3rd (Conv1D) (None, 127, 256) 65792 leaky_re_lu[19][0] __________________________________________________________________________________________________ shared_res_13_3rd_batchnorm (Ba (None, 127, 256) 1024 shared_res_13_3rd[0][0] __________________________________________________________________________________________________ add_5 (Add) (None, 127, 256) 0 shared_res_13_3rd_batchnorm[0][0] max_pooling1d_4[0][0] __________________________________________________________________________________________________ max_pooling1d_5 (MaxPooling1D) (None, 63, 256) 0 leaky_re_lu[20][0] __________________________________________________________________________________________________ shared_conv_15 (Conv1D) (None, 63, 512) 393728 max_pooling1d_5[0][0] __________________________________________________________________________________________________ shared_conv_15_batchnorm (Batch (None, 63, 512) 2048 shared_conv_15[0][0] __________________________________________________________________________________________________ shared_res_16_1st (Conv1D) (None, 63, 512) 262656 leaky_re_lu[21][0] __________________________________________________________________________________________________ shared_res_16_1st_batchnorm (Ba (None, 63, 512) 2048 shared_res_16_1st[0][0] __________________________________________________________________________________________________ shared_res_16_2nd (Conv1D) (None, 63, 512) 786944 leaky_re_lu[22][0] __________________________________________________________________________________________________ shared_res_16_2nd_batchnorm (Ba (None, 63, 512) 2048 shared_res_16_2nd[0][0] __________________________________________________________________________________________________ shared_res_16_3rd (Conv1D) (None, 63, 512) 262656 leaky_re_lu[23][0] __________________________________________________________________________________________________ shared_res_16_3rd_batchnorm (Ba (None, 63, 512) 2048 shared_res_16_3rd[0][0] __________________________________________________________________________________________________ add_6 (Add) (None, 63, 512) 0 shared_res_16_3rd_batchnorm[0][0] leaky_re_lu[21][0] __________________________________________________________________________________________________ max_pooling1d_6 (MaxPooling1D) (None, 31, 512) 0 leaky_re_lu[24][0] __________________________________________________________________________________________________ shared_res_18_1st (Conv1D) (None, 31, 512) 262656 max_pooling1d_6[0][0] __________________________________________________________________________________________________ shared_res_18_1st_batchnorm (Ba (None, 31, 512) 2048 shared_res_18_1st[0][0] __________________________________________________________________________________________________ shared_res_18_2nd (Conv1D) (None, 31, 512) 786944 leaky_re_lu[25][0] __________________________________________________________________________________________________ shared_res_18_2nd_batchnorm (Ba (None, 31, 512) 2048 shared_res_18_2nd[0][0] __________________________________________________________________________________________________ shared_res_18_3rd (Conv1D) (None, 31, 512) 262656 leaky_re_lu[26][0] __________________________________________________________________________________________________ shared_res_18_3rd_batchnorm (Ba (None, 31, 512) 2048 shared_res_18_3rd[0][0] __________________________________________________________________________________________________ add_7 (Add) (None, 31, 512) 0 shared_res_18_3rd_batchnorm[0][0] max_pooling1d_6[0][0] __________________________________________________________________________________________________ max_pooling1d_7 (MaxPooling1D) (None, 15, 512) 0 leaky_re_lu[27][0] __________________________________________________________________________________________________ shared_conv_20 (Conv1D) (None, 15, 1024) 525312 max_pooling1d_7[0][0] __________________________________________________________________________________________________ shared_conv_20_batchnorm (Batch (None, 15, 1024) 4096 shared_conv_20[0][0] __________________________________________________________________________________________________ shared_res_21_1st (Conv1D) (None, 15, 1024) 1049600 leaky_re_lu[28][0] __________________________________________________________________________________________________ shared_res_21_1st_batchnorm (Ba (None, 15, 1024) 4096 shared_res_21_1st[0][0] __________________________________________________________________________________________________ shared_res_21_2nd (Conv1D) (None, 15, 1024) 1049600 leaky_re_lu[29][0] __________________________________________________________________________________________________ shared_res_21_2nd_batchnorm (Ba (None, 15, 1024) 4096 shared_res_21_2nd[0][0] __________________________________________________________________________________________________ shared_res_21_3rd (Conv1D) (None, 15, 1024) 1049600 leaky_re_lu[30][0] __________________________________________________________________________________________________ shared_res_21_3rd_batchnorm (Ba (None, 15, 1024) 4096 shared_res_21_3rd[0][0] __________________________________________________________________________________________________ add_8 (Add) (None, 15, 1024) 0 shared_res_21_3rd_batchnorm[0][0] leaky_re_lu[28][0] __________________________________________________________________________________________________ max_pooling1d_8 (MaxPooling1D) (None, 7, 1024) 0 leaky_re_lu[31][0] __________________________________________________________________________________________________ shared_res_23_1st (Conv1D) (None, 7, 1024) 1049600 max_pooling1d_8[0][0] __________________________________________________________________________________________________ shared_res_23_1st_batchnorm (Ba (None, 7, 1024) 4096 shared_res_23_1st[0][0] __________________________________________________________________________________________________ shared_res_23_2nd (Conv1D) (None, 7, 1024) 1049600 leaky_re_lu[32][0] __________________________________________________________________________________________________ shared_res_23_2nd_batchnorm (Ba (None, 7, 1024) 4096 shared_res_23_2nd[0][0] __________________________________________________________________________________________________ shared_res_23_3rd (Conv1D) (None, 7, 1024) 1049600 leaky_re_lu[33][0] __________________________________________________________________________________________________ shared_res_23_3rd_batchnorm (Ba (None, 7, 1024) 4096 shared_res_23_3rd[0][0] __________________________________________________________________________________________________ add_9 (Add) (None, 7, 1024) 0 shared_res_23_3rd_batchnorm[0][0] max_pooling1d_8[0][0] __________________________________________________________________________________________________ max_pooling1d_9 (MaxPooling1D) (None, 3, 1024) 0 leaky_re_lu[34][0] __________________________________________________________________________________________________ flatten (Flatten) (None, 3072) 0 max_pooling1d_9[0][0] __________________________________________________________________________________________________ r1_dense_1 (Dense) (None, 1024) 3146752 flatten[0][0] __________________________________________________________________________________________________ r1_dense_1_batchnorm (BatchNorm (None, 1024) 4096 r1_dense_1[0][0] __________________________________________________________________________________________________ r1_dense_2 (Dense) (None, 1024) 1049600 leaky_re_lu[35][0] __________________________________________________________________________________________________ r1_dense_2_batchnorm (BatchNorm (None, 1024) 4096 r1_dense_2[0][0] __________________________________________________________________________________________________ r1_dense_3 (Dense) (None, 1024) 1049600 leaky_re_lu[36][0] __________________________________________________________________________________________________ r1_dense_3_batchnorm (BatchNorm (None, 1024) 4096 r1_dense_3[0][0] __________________________________________________________________________________________________ r1_dense_4 (Dense) (None, 1024) 1049600 leaky_re_lu[37][0] __________________________________________________________________________________________________ r1_dense_4_batchnorm (BatchNorm (None, 1024) 4096 r1_dense_4[0][0] __________________________________________________________________________________________________ r1_mean_dense (Dense) (None, 12) 12300 leaky_re_lu[38][0] __________________________________________________________________________________________________ r1_logvar_dense (Dense) (None, 12) 12300 leaky_re_lu[38][0] __________________________________________________________________________________________________ r1_modes_dense (Dense) (None, 6) 6150 leaky_re_lu[38][0] __________________________________________________________________________________________________ concatenate (Concatenate) (None, 30) 0 r1_mean_dense[0][0] r1_logvar_dense[0][0] r1_modes_dense[0][0] ================================================================================================== Total params: 18,647,070 Trainable params: 18,611,102 Non-trainable params: 35,968 __________________________________________________________________________________________________ Model: "encoder_q" __________________________________________________________________________________________________ Layer (type) Output Shape Param # Connected to ================================================================================================== input_1 (InputLayer) [(None, 4096, 3)] 0 __________________________________________________________________________________________________ shared_conv_0 (Conv1D) (None, 4096, 64) 12544 input_1[0][0] __________________________________________________________________________________________________ shared_conv_0_batchnorm (BatchN (None, 4096, 64) 256 shared_conv_0[0][0] __________________________________________________________________________________________________ leaky_re_lu (LeakyReLU) multiple 0 shared_conv_0_batchnorm[0][0] shared_res_1_1st_batchnorm[0][0] shared_res_1_2nd_batchnorm[0][0] add[0][0] shared_res_3_1st_batchnorm[0][0] shared_res_3_2nd_batchnorm[0][0] add_1[0][0] shared_conv_5_batchnorm[0][0] shared_res_6_1st_batchnorm[0][0] shared_res_6_2nd_batchnorm[0][0] add_2[0][0] shared_res_8_1st_batchnorm[0][0] shared_res_8_2nd_batchnorm[0][0] add_3[0][0] shared_conv_10_batchnorm[0][0] shared_res_11_1st_batchnorm[0][0] shared_res_11_2nd_batchnorm[0][0] add_4[0][0] shared_res_13_1st_batchnorm[0][0] shared_res_13_2nd_batchnorm[0][0] add_5[0][0] shared_conv_15_batchnorm[0][0] shared_res_16_1st_batchnorm[0][0] shared_res_16_2nd_batchnorm[0][0] add_6[0][0] shared_res_18_1st_batchnorm[0][0] shared_res_18_2nd_batchnorm[0][0] add_7[0][0] shared_conv_20_batchnorm[0][0] shared_res_21_1st_batchnorm[0][0] shared_res_21_2nd_batchnorm[0][0] add_8[0][0] shared_res_23_1st_batchnorm[0][0] shared_res_23_2nd_batchnorm[0][0] add_9[0][0] batch_normalization[0][0] q_dense_1_batchnorm[0][0] q_dense_2_batchnorm[0][0] q_dense_3_batchnorm[0][0] q_dense_4_batchnorm[0][0] __________________________________________________________________________________________________ shared_res_1_1st (Conv1D) (None, 4096, 64) 4160 leaky_re_lu[0][0] __________________________________________________________________________________________________ shared_res_1_1st_batchnorm (Bat (None, 4096, 64) 256 shared_res_1_1st[0][0] __________________________________________________________________________________________________ shared_res_1_2nd (Conv1D) (None, 4096, 64) 266304 leaky_re_lu[1][0] __________________________________________________________________________________________________ shared_res_1_2nd_batchnorm (Bat (None, 4096, 64) 256 shared_res_1_2nd[0][0] __________________________________________________________________________________________________ shared_res_1_3rd (Conv1D) (None, 4096, 64) 4160 leaky_re_lu[2][0] __________________________________________________________________________________________________ shared_res_1_3rd_batchnorm (Bat (None, 4096, 64) 256 shared_res_1_3rd[0][0] __________________________________________________________________________________________________ add (Add) (None, 4096, 64) 0 shared_res_1_3rd_batchnorm[0][0] leaky_re_lu[0][0] __________________________________________________________________________________________________ max_pooling1d (MaxPooling1D) (None, 2047, 64) 0 leaky_re_lu[3][0] __________________________________________________________________________________________________ shared_res_3_1st (Conv1D) (None, 2047, 64) 4160 max_pooling1d[0][0] __________________________________________________________________________________________________ shared_res_3_1st_batchnorm (Bat (None, 2047, 64) 256 shared_res_3_1st[0][0] __________________________________________________________________________________________________ shared_res_3_2nd (Conv1D) (None, 2047, 64) 266304 leaky_re_lu[4][0] __________________________________________________________________________________________________ shared_res_3_2nd_batchnorm (Bat (None, 2047, 64) 256 shared_res_3_2nd[0][0] __________________________________________________________________________________________________ shared_res_3_3rd (Conv1D) (None, 2047, 64) 4160 leaky_re_lu[5][0] __________________________________________________________________________________________________ shared_res_3_3rd_batchnorm (Bat (None, 2047, 64) 256 shared_res_3_3rd[0][0] __________________________________________________________________________________________________ add_1 (Add) (None, 2047, 64) 0 shared_res_3_3rd_batchnorm[0][0] max_pooling1d[0][0] __________________________________________________________________________________________________ max_pooling1d_1 (MaxPooling1D) (None, 1023, 64) 0 leaky_re_lu[6][0] __________________________________________________________________________________________________ shared_conv_5 (Conv1D) (None, 1023, 128) 139392 max_pooling1d_1[0][0] __________________________________________________________________________________________________ shared_conv_5_batchnorm (BatchN (None, 1023, 128) 512 shared_conv_5[0][0] __________________________________________________________________________________________________ shared_res_6_1st (Conv1D) (None, 1023, 128) 16512 leaky_re_lu[7][0] __________________________________________________________________________________________________ shared_res_6_1st_batchnorm (Bat (None, 1023, 128) 512 shared_res_6_1st[0][0] __________________________________________________________________________________________________ shared_res_6_2nd (Conv1D) (None, 1023, 128) 278656 leaky_re_lu[8][0] __________________________________________________________________________________________________ shared_res_6_2nd_batchnorm (Bat (None, 1023, 128) 512 shared_res_6_2nd[0][0] __________________________________________________________________________________________________ shared_res_6_3rd (Conv1D) (None, 1023, 128) 16512 leaky_re_lu[9][0] __________________________________________________________________________________________________ shared_res_6_3rd_batchnorm (Bat (None, 1023, 128) 512 shared_res_6_3rd[0][0] __________________________________________________________________________________________________ add_2 (Add) (None, 1023, 128) 0 shared_res_6_3rd_batchnorm[0][0] leaky_re_lu[7][0] __________________________________________________________________________________________________ max_pooling1d_2 (MaxPooling1D) (None, 511, 128) 0 leaky_re_lu[10][0] __________________________________________________________________________________________________ shared_res_8_1st (Conv1D) (None, 511, 128) 16512 max_pooling1d_2[0][0] __________________________________________________________________________________________________ shared_res_8_1st_batchnorm (Bat (None, 511, 128) 512 shared_res_8_1st[0][0] __________________________________________________________________________________________________ shared_res_8_2nd (Conv1D) (None, 511, 128) 278656 leaky_re_lu[11][0] __________________________________________________________________________________________________ shared_res_8_2nd_batchnorm (Bat (None, 511, 128) 512 shared_res_8_2nd[0][0] __________________________________________________________________________________________________ shared_res_8_3rd (Conv1D) (None, 511, 128) 16512 leaky_re_lu[12][0] __________________________________________________________________________________________________ shared_res_8_3rd_batchnorm (Bat (None, 511, 128) 512 shared_res_8_3rd[0][0] __________________________________________________________________________________________________ add_3 (Add) (None, 511, 128) 0 shared_res_8_3rd_batchnorm[0][0] max_pooling1d_2[0][0] __________________________________________________________________________________________________ max_pooling1d_3 (MaxPooling1D) (None, 255, 128) 0 leaky_re_lu[13][0] __________________________________________________________________________________________________ shared_conv_10 (Conv1D) (None, 255, 256) 164096 max_pooling1d_3[0][0] __________________________________________________________________________________________________ shared_conv_10_batchnorm (Batch (None, 255, 256) 1024 shared_conv_10[0][0] __________________________________________________________________________________________________ shared_res_11_1st (Conv1D) (None, 255, 256) 65792 leaky_re_lu[14][0] __________________________________________________________________________________________________ shared_res_11_1st_batchnorm (Ba (None, 255, 256) 1024 shared_res_11_1st[0][0] __________________________________________________________________________________________________ shared_res_11_2nd (Conv1D) (None, 255, 256) 327936 leaky_re_lu[15][0] __________________________________________________________________________________________________ shared_res_11_2nd_batchnorm (Ba (None, 255, 256) 1024 shared_res_11_2nd[0][0] __________________________________________________________________________________________________ shared_res_11_3rd (Conv1D) (None, 255, 256) 65792 leaky_re_lu[16][0] __________________________________________________________________________________________________ shared_res_11_3rd_batchnorm (Ba (None, 255, 256) 1024 shared_res_11_3rd[0][0] __________________________________________________________________________________________________ add_4 (Add) (None, 255, 256) 0 shared_res_11_3rd_batchnorm[0][0] leaky_re_lu[14][0] __________________________________________________________________________________________________ max_pooling1d_4 (MaxPooling1D) (None, 127, 256) 0 leaky_re_lu[17][0] __________________________________________________________________________________________________ shared_res_13_1st (Conv1D) (None, 127, 256) 65792 max_pooling1d_4[0][0] __________________________________________________________________________________________________ shared_res_13_1st_batchnorm (Ba (None, 127, 256) 1024 shared_res_13_1st[0][0] __________________________________________________________________________________________________ shared_res_13_2nd (Conv1D) (None, 127, 256) 327936 leaky_re_lu[18][0] __________________________________________________________________________________________________ shared_res_13_2nd_batchnorm (Ba (None, 127, 256) 1024 shared_res_13_2nd[0][0] __________________________________________________________________________________________________ shared_res_13_3rd (Conv1D) (None, 127, 256) 65792 leaky_re_lu[19][0] __________________________________________________________________________________________________ shared_res_13_3rd_batchnorm (Ba (None, 127, 256) 1024 shared_res_13_3rd[0][0] __________________________________________________________________________________________________ add_5 (Add) (None, 127, 256) 0 shared_res_13_3rd_batchnorm[0][0] max_pooling1d_4[0][0] __________________________________________________________________________________________________ max_pooling1d_5 (MaxPooling1D) (None, 63, 256) 0 leaky_re_lu[20][0] __________________________________________________________________________________________________ shared_conv_15 (Conv1D) (None, 63, 512) 393728 max_pooling1d_5[0][0] __________________________________________________________________________________________________ shared_conv_15_batchnorm (Batch (None, 63, 512) 2048 shared_conv_15[0][0] __________________________________________________________________________________________________ shared_res_16_1st (Conv1D) (None, 63, 512) 262656 leaky_re_lu[21][0] __________________________________________________________________________________________________ shared_res_16_1st_batchnorm (Ba (None, 63, 512) 2048 shared_res_16_1st[0][0] __________________________________________________________________________________________________ shared_res_16_2nd (Conv1D) (None, 63, 512) 786944 leaky_re_lu[22][0] __________________________________________________________________________________________________ shared_res_16_2nd_batchnorm (Ba (None, 63, 512) 2048 shared_res_16_2nd[0][0] __________________________________________________________________________________________________ shared_res_16_3rd (Conv1D) (None, 63, 512) 262656 leaky_re_lu[23][0] __________________________________________________________________________________________________ shared_res_16_3rd_batchnorm (Ba (None, 63, 512) 2048 shared_res_16_3rd[0][0] __________________________________________________________________________________________________ add_6 (Add) (None, 63, 512) 0 shared_res_16_3rd_batchnorm[0][0] leaky_re_lu[21][0] __________________________________________________________________________________________________ max_pooling1d_6 (MaxPooling1D) (None, 31, 512) 0 leaky_re_lu[24][0] __________________________________________________________________________________________________ shared_res_18_1st (Conv1D) (None, 31, 512) 262656 max_pooling1d_6[0][0] __________________________________________________________________________________________________ shared_res_18_1st_batchnorm (Ba (None, 31, 512) 2048 shared_res_18_1st[0][0] __________________________________________________________________________________________________ shared_res_18_2nd (Conv1D) (None, 31, 512) 786944 leaky_re_lu[25][0] __________________________________________________________________________________________________ shared_res_18_2nd_batchnorm (Ba (None, 31, 512) 2048 shared_res_18_2nd[0][0] __________________________________________________________________________________________________ shared_res_18_3rd (Conv1D) (None, 31, 512) 262656 leaky_re_lu[26][0] __________________________________________________________________________________________________ shared_res_18_3rd_batchnorm (Ba (None, 31, 512) 2048 shared_res_18_3rd[0][0] __________________________________________________________________________________________________ add_7 (Add) (None, 31, 512) 0 shared_res_18_3rd_batchnorm[0][0] max_pooling1d_6[0][0] __________________________________________________________________________________________________ max_pooling1d_7 (MaxPooling1D) (None, 15, 512) 0 leaky_re_lu[27][0] __________________________________________________________________________________________________ shared_conv_20 (Conv1D) (None, 15, 1024) 525312 max_pooling1d_7[0][0] __________________________________________________________________________________________________ shared_conv_20_batchnorm (Batch (None, 15, 1024) 4096 shared_conv_20[0][0] __________________________________________________________________________________________________ shared_res_21_1st (Conv1D) (None, 15, 1024) 1049600 leaky_re_lu[28][0] __________________________________________________________________________________________________ shared_res_21_1st_batchnorm (Ba (None, 15, 1024) 4096 shared_res_21_1st[0][0] __________________________________________________________________________________________________ shared_res_21_2nd (Conv1D) (None, 15, 1024) 1049600 leaky_re_lu[29][0] __________________________________________________________________________________________________ shared_res_21_2nd_batchnorm (Ba (None, 15, 1024) 4096 shared_res_21_2nd[0][0] __________________________________________________________________________________________________ shared_res_21_3rd (Conv1D) (None, 15, 1024) 1049600 leaky_re_lu[30][0] __________________________________________________________________________________________________ shared_res_21_3rd_batchnorm (Ba (None, 15, 1024) 4096 shared_res_21_3rd[0][0] __________________________________________________________________________________________________ add_8 (Add) (None, 15, 1024) 0 shared_res_21_3rd_batchnorm[0][0] leaky_re_lu[28][0] __________________________________________________________________________________________________ max_pooling1d_8 (MaxPooling1D) (None, 7, 1024) 0 leaky_re_lu[31][0] __________________________________________________________________________________________________ shared_res_23_1st (Conv1D) (None, 7, 1024) 1049600 max_pooling1d_8[0][0] __________________________________________________________________________________________________ shared_res_23_1st_batchnorm (Ba (None, 7, 1024) 4096 shared_res_23_1st[0][0] __________________________________________________________________________________________________ shared_res_23_2nd (Conv1D) (None, 7, 1024) 1049600 leaky_re_lu[32][0] __________________________________________________________________________________________________ shared_res_23_2nd_batchnorm (Ba (None, 7, 1024) 4096 shared_res_23_2nd[0][0] __________________________________________________________________________________________________ shared_res_23_3rd (Conv1D) (None, 7, 1024) 1049600 leaky_re_lu[33][0] __________________________________________________________________________________________________ input_2 (InputLayer) [(None, 2)] 0 __________________________________________________________________________________________________ shared_res_23_3rd_batchnorm (Ba (None, 7, 1024) 4096 shared_res_23_3rd[0][0] __________________________________________________________________________________________________ flatten_1 (Flatten) (None, 2) 0 input_2[0][0] __________________________________________________________________________________________________ add_9 (Add) (None, 7, 1024) 0 shared_res_23_3rd_batchnorm[0][0] max_pooling1d_8[0][0] __________________________________________________________________________________________________ q_inx_dense (Dense) (None, 3072) 9216 flatten_1[0][0] __________________________________________________________________________________________________ tf.reshape (TFOpLambda) (None, 3, 1024, 1) 0 q_inx_dense[0][0] __________________________________________________________________________________________________ max_pooling1d_9 (MaxPooling1D) (None, 3, 1024) 0 leaky_re_lu[34][0] __________________________________________________________________________________________________ batch_normalization (BatchNorma (None, 3, 1024, 1) 4 tf.reshape[0][0] __________________________________________________________________________________________________ tf.expand_dims (TFOpLambda) (None, 3, 1024, 1) 0 max_pooling1d_9[0][0] __________________________________________________________________________________________________ concatenate_1 (Concatenate) (None, 3, 1024, 2) 0 tf.expand_dims[0][0] leaky_re_lu[39][0] __________________________________________________________________________________________________ flatten_2 (Flatten) (None, 6144) 0 concatenate_1[0][0] __________________________________________________________________________________________________ q_dense_1 (Dense) (None, 1024) 6292480 flatten_2[0][0] __________________________________________________________________________________________________ q_dense_1_batchnorm (BatchNorma (None, 1024) 4096 q_dense_1[0][0] __________________________________________________________________________________________________ q_dense_2 (Dense) (None, 1024) 1049600 leaky_re_lu[40][0] __________________________________________________________________________________________________ q_dense_2_batchnorm (BatchNorma (None, 1024) 4096 q_dense_2[0][0] __________________________________________________________________________________________________ q_dense_3 (Dense) (None, 1024) 1049600 leaky_re_lu[41][0] __________________________________________________________________________________________________ q_dense_3_batchnorm (BatchNorma (None, 1024) 4096 q_dense_3[0][0] __________________________________________________________________________________________________ q_dense_4 (Dense) (None, 1024) 1049600 leaky_re_lu[42][0] __________________________________________________________________________________________________ q_dense_4_batchnorm (BatchNorma (None, 1024) 4096 q_dense_4[0][0] __________________________________________________________________________________________________ q_mean_dense (Dense) (None, 4) 4100 leaky_re_lu[43][0] __________________________________________________________________________________________________ q_logvar_dense (Dense) (None, 6) 6150 leaky_re_lu[43][0] __________________________________________________________________________________________________ q_modes_dense (Dense) (None, 2) 2050 leaky_re_lu[43][0] __________________________________________________________________________________________________ concatenate_2 (Concatenate) (None, 12) 0 q_mean_dense[0][0] q_logvar_dense[0][0] q_modes_dense[0][0] ================================================================================================== Total params: 21,783,568 Trainable params: 21,747,598 Non-trainable params: 35,970 __________________________________________________________________________________________________ Model: "decoder_r2" __________________________________________________________________________________________________ Layer (type) Output Shape Param # Connected to ================================================================================================== input_1 (InputLayer) [(None, 4096, 3)] 0 __________________________________________________________________________________________________ shared_conv_0 (Conv1D) (None, 4096, 64) 12544 input_1[0][0] __________________________________________________________________________________________________ shared_conv_0_batchnorm (BatchN (None, 4096, 64) 256 shared_conv_0[0][0] __________________________________________________________________________________________________ leaky_re_lu (LeakyReLU) multiple 0 shared_conv_0_batchnorm[0][0] shared_res_1_1st_batchnorm[0][0] shared_res_1_2nd_batchnorm[0][0] add[0][0] shared_res_3_1st_batchnorm[0][0] shared_res_3_2nd_batchnorm[0][0] add_1[0][0] shared_conv_5_batchnorm[0][0] shared_res_6_1st_batchnorm[0][0] shared_res_6_2nd_batchnorm[0][0] add_2[0][0] shared_res_8_1st_batchnorm[0][0] shared_res_8_2nd_batchnorm[0][0] add_3[0][0] shared_conv_10_batchnorm[0][0] shared_res_11_1st_batchnorm[0][0] shared_res_11_2nd_batchnorm[0][0] add_4[0][0] shared_res_13_1st_batchnorm[0][0] shared_res_13_2nd_batchnorm[0][0] add_5[0][0] shared_conv_15_batchnorm[0][0] shared_res_16_1st_batchnorm[0][0] shared_res_16_2nd_batchnorm[0][0] add_6[0][0] shared_res_18_1st_batchnorm[0][0] shared_res_18_2nd_batchnorm[0][0] add_7[0][0] shared_conv_20_batchnorm[0][0] shared_res_21_1st_batchnorm[0][0] shared_res_21_2nd_batchnorm[0][0] add_8[0][0] shared_res_23_1st_batchnorm[0][0] shared_res_23_2nd_batchnorm[0][0] add_9[0][0] batch_normalization_1[0][0] r2_dense_1_batchnorm[0][0] r2_dense_2_batchnorm[0][0] r2_dense_3_batchnorm[0][0] r2_dense_4_batchnorm[0][0] __________________________________________________________________________________________________ shared_res_1_1st (Conv1D) (None, 4096, 64) 4160 leaky_re_lu[0][0] __________________________________________________________________________________________________ shared_res_1_1st_batchnorm (Bat (None, 4096, 64) 256 shared_res_1_1st[0][0] __________________________________________________________________________________________________ shared_res_1_2nd (Conv1D) (None, 4096, 64) 266304 leaky_re_lu[1][0] __________________________________________________________________________________________________ shared_res_1_2nd_batchnorm (Bat (None, 4096, 64) 256 shared_res_1_2nd[0][0] __________________________________________________________________________________________________ shared_res_1_3rd (Conv1D) (None, 4096, 64) 4160 leaky_re_lu[2][0] __________________________________________________________________________________________________ shared_res_1_3rd_batchnorm (Bat (None, 4096, 64) 256 shared_res_1_3rd[0][0] __________________________________________________________________________________________________ add (Add) (None, 4096, 64) 0 shared_res_1_3rd_batchnorm[0][0] leaky_re_lu[0][0] __________________________________________________________________________________________________ max_pooling1d (MaxPooling1D) (None, 2047, 64) 0 leaky_re_lu[3][0] __________________________________________________________________________________________________ shared_res_3_1st (Conv1D) (None, 2047, 64) 4160 max_pooling1d[0][0] __________________________________________________________________________________________________ shared_res_3_1st_batchnorm (Bat (None, 2047, 64) 256 shared_res_3_1st[0][0] __________________________________________________________________________________________________ shared_res_3_2nd (Conv1D) (None, 2047, 64) 266304 leaky_re_lu[4][0] __________________________________________________________________________________________________ shared_res_3_2nd_batchnorm (Bat (None, 2047, 64) 256 shared_res_3_2nd[0][0] __________________________________________________________________________________________________ shared_res_3_3rd (Conv1D) (None, 2047, 64) 4160 leaky_re_lu[5][0] __________________________________________________________________________________________________ shared_res_3_3rd_batchnorm (Bat (None, 2047, 64) 256 shared_res_3_3rd[0][0] __________________________________________________________________________________________________ add_1 (Add) (None, 2047, 64) 0 shared_res_3_3rd_batchnorm[0][0] max_pooling1d[0][0] __________________________________________________________________________________________________ max_pooling1d_1 (MaxPooling1D) (None, 1023, 64) 0 leaky_re_lu[6][0] __________________________________________________________________________________________________ shared_conv_5 (Conv1D) (None, 1023, 128) 139392 max_pooling1d_1[0][0] __________________________________________________________________________________________________ shared_conv_5_batchnorm (BatchN (None, 1023, 128) 512 shared_conv_5[0][0] __________________________________________________________________________________________________ shared_res_6_1st (Conv1D) (None, 1023, 128) 16512 leaky_re_lu[7][0] __________________________________________________________________________________________________ shared_res_6_1st_batchnorm (Bat (None, 1023, 128) 512 shared_res_6_1st[0][0] __________________________________________________________________________________________________ shared_res_6_2nd (Conv1D) (None, 1023, 128) 278656 leaky_re_lu[8][0] __________________________________________________________________________________________________ shared_res_6_2nd_batchnorm (Bat (None, 1023, 128) 512 shared_res_6_2nd[0][0] __________________________________________________________________________________________________ shared_res_6_3rd (Conv1D) (None, 1023, 128) 16512 leaky_re_lu[9][0] __________________________________________________________________________________________________ shared_res_6_3rd_batchnorm (Bat (None, 1023, 128) 512 shared_res_6_3rd[0][0] __________________________________________________________________________________________________ add_2 (Add) (None, 1023, 128) 0 shared_res_6_3rd_batchnorm[0][0] leaky_re_lu[7][0] __________________________________________________________________________________________________ max_pooling1d_2 (MaxPooling1D) (None, 511, 128) 0 leaky_re_lu[10][0] __________________________________________________________________________________________________ shared_res_8_1st (Conv1D) (None, 511, 128) 16512 max_pooling1d_2[0][0] __________________________________________________________________________________________________ shared_res_8_1st_batchnorm (Bat (None, 511, 128) 512 shared_res_8_1st[0][0] __________________________________________________________________________________________________ shared_res_8_2nd (Conv1D) (None, 511, 128) 278656 leaky_re_lu[11][0] __________________________________________________________________________________________________ shared_res_8_2nd_batchnorm (Bat (None, 511, 128) 512 shared_res_8_2nd[0][0] __________________________________________________________________________________________________ shared_res_8_3rd (Conv1D) (None, 511, 128) 16512 leaky_re_lu[12][0] __________________________________________________________________________________________________ shared_res_8_3rd_batchnorm (Bat (None, 511, 128) 512 shared_res_8_3rd[0][0] __________________________________________________________________________________________________ add_3 (Add) (None, 511, 128) 0 shared_res_8_3rd_batchnorm[0][0] max_pooling1d_2[0][0] __________________________________________________________________________________________________ max_pooling1d_3 (MaxPooling1D) (None, 255, 128) 0 leaky_re_lu[13][0] __________________________________________________________________________________________________ shared_conv_10 (Conv1D) (None, 255, 256) 164096 max_pooling1d_3[0][0] __________________________________________________________________________________________________ shared_conv_10_batchnorm (Batch (None, 255, 256) 1024 shared_conv_10[0][0] __________________________________________________________________________________________________ shared_res_11_1st (Conv1D) (None, 255, 256) 65792 leaky_re_lu[14][0] __________________________________________________________________________________________________ shared_res_11_1st_batchnorm (Ba (None, 255, 256) 1024 shared_res_11_1st[0][0] __________________________________________________________________________________________________ shared_res_11_2nd (Conv1D) (None, 255, 256) 327936 leaky_re_lu[15][0] __________________________________________________________________________________________________ shared_res_11_2nd_batchnorm (Ba (None, 255, 256) 1024 shared_res_11_2nd[0][0] __________________________________________________________________________________________________ shared_res_11_3rd (Conv1D) (None, 255, 256) 65792 leaky_re_lu[16][0] __________________________________________________________________________________________________ shared_res_11_3rd_batchnorm (Ba (None, 255, 256) 1024 shared_res_11_3rd[0][0] __________________________________________________________________________________________________ add_4 (Add) (None, 255, 256) 0 shared_res_11_3rd_batchnorm[0][0] leaky_re_lu[14][0] __________________________________________________________________________________________________ max_pooling1d_4 (MaxPooling1D) (None, 127, 256) 0 leaky_re_lu[17][0] __________________________________________________________________________________________________ shared_res_13_1st (Conv1D) (None, 127, 256) 65792 max_pooling1d_4[0][0] __________________________________________________________________________________________________ shared_res_13_1st_batchnorm (Ba (None, 127, 256) 1024 shared_res_13_1st[0][0] __________________________________________________________________________________________________ shared_res_13_2nd (Conv1D) (None, 127, 256) 327936 leaky_re_lu[18][0] __________________________________________________________________________________________________ shared_res_13_2nd_batchnorm (Ba (None, 127, 256) 1024 shared_res_13_2nd[0][0] __________________________________________________________________________________________________ shared_res_13_3rd (Conv1D) (None, 127, 256) 65792 leaky_re_lu[19][0] __________________________________________________________________________________________________ shared_res_13_3rd_batchnorm (Ba (None, 127, 256) 1024 shared_res_13_3rd[0][0] __________________________________________________________________________________________________ add_5 (Add) (None, 127, 256) 0 shared_res_13_3rd_batchnorm[0][0] max_pooling1d_4[0][0] __________________________________________________________________________________________________ max_pooling1d_5 (MaxPooling1D) (None, 63, 256) 0 leaky_re_lu[20][0] __________________________________________________________________________________________________ shared_conv_15 (Conv1D) (None, 63, 512) 393728 max_pooling1d_5[0][0] __________________________________________________________________________________________________ shared_conv_15_batchnorm (Batch (None, 63, 512) 2048 shared_conv_15[0][0] __________________________________________________________________________________________________ shared_res_16_1st (Conv1D) (None, 63, 512) 262656 leaky_re_lu[21][0] __________________________________________________________________________________________________ shared_res_16_1st_batchnorm (Ba (None, 63, 512) 2048 shared_res_16_1st[0][0] __________________________________________________________________________________________________ shared_res_16_2nd (Conv1D) (None, 63, 512) 786944 leaky_re_lu[22][0] __________________________________________________________________________________________________ shared_res_16_2nd_batchnorm (Ba (None, 63, 512) 2048 shared_res_16_2nd[0][0] __________________________________________________________________________________________________ shared_res_16_3rd (Conv1D) (None, 63, 512) 262656 leaky_re_lu[23][0] __________________________________________________________________________________________________ shared_res_16_3rd_batchnorm (Ba (None, 63, 512) 2048 shared_res_16_3rd[0][0] __________________________________________________________________________________________________ add_6 (Add) (None, 63, 512) 0 shared_res_16_3rd_batchnorm[0][0] leaky_re_lu[21][0] __________________________________________________________________________________________________ max_pooling1d_6 (MaxPooling1D) (None, 31, 512) 0 leaky_re_lu[24][0] __________________________________________________________________________________________________ shared_res_18_1st (Conv1D) (None, 31, 512) 262656 max_pooling1d_6[0][0] __________________________________________________________________________________________________ shared_res_18_1st_batchnorm (Ba (None, 31, 512) 2048 shared_res_18_1st[0][0] __________________________________________________________________________________________________ shared_res_18_2nd (Conv1D) (None, 31, 512) 786944 leaky_re_lu[25][0] __________________________________________________________________________________________________ shared_res_18_2nd_batchnorm (Ba (None, 31, 512) 2048 shared_res_18_2nd[0][0] __________________________________________________________________________________________________ shared_res_18_3rd (Conv1D) (None, 31, 512) 262656 leaky_re_lu[26][0] __________________________________________________________________________________________________ shared_res_18_3rd_batchnorm (Ba (None, 31, 512) 2048 shared_res_18_3rd[0][0] __________________________________________________________________________________________________ add_7 (Add) (None, 31, 512) 0 shared_res_18_3rd_batchnorm[0][0] max_pooling1d_6[0][0] __________________________________________________________________________________________________ max_pooling1d_7 (MaxPooling1D) (None, 15, 512) 0 leaky_re_lu[27][0] __________________________________________________________________________________________________ shared_conv_20 (Conv1D) (None, 15, 1024) 525312 max_pooling1d_7[0][0] __________________________________________________________________________________________________ shared_conv_20_batchnorm (Batch (None, 15, 1024) 4096 shared_conv_20[0][0] __________________________________________________________________________________________________ shared_res_21_1st (Conv1D) (None, 15, 1024) 1049600 leaky_re_lu[28][0] __________________________________________________________________________________________________ shared_res_21_1st_batchnorm (Ba (None, 15, 1024) 4096 shared_res_21_1st[0][0] __________________________________________________________________________________________________ shared_res_21_2nd (Conv1D) (None, 15, 1024) 1049600 leaky_re_lu[29][0] __________________________________________________________________________________________________ shared_res_21_2nd_batchnorm (Ba (None, 15, 1024) 4096 shared_res_21_2nd[0][0] __________________________________________________________________________________________________ shared_res_21_3rd (Conv1D) (None, 15, 1024) 1049600 leaky_re_lu[30][0] __________________________________________________________________________________________________ shared_res_21_3rd_batchnorm (Ba (None, 15, 1024) 4096 shared_res_21_3rd[0][0] __________________________________________________________________________________________________ add_8 (Add) (None, 15, 1024) 0 shared_res_21_3rd_batchnorm[0][0] leaky_re_lu[28][0] __________________________________________________________________________________________________ max_pooling1d_8 (MaxPooling1D) (None, 7, 1024) 0 leaky_re_lu[31][0] __________________________________________________________________________________________________ shared_res_23_1st (Conv1D) (None, 7, 1024) 1049600 max_pooling1d_8[0][0] __________________________________________________________________________________________________ shared_res_23_1st_batchnorm (Ba (None, 7, 1024) 4096 shared_res_23_1st[0][0] __________________________________________________________________________________________________ shared_res_23_2nd (Conv1D) (None, 7, 1024) 1049600 leaky_re_lu[32][0] __________________________________________________________________________________________________ shared_res_23_2nd_batchnorm (Ba (None, 7, 1024) 4096 shared_res_23_2nd[0][0] __________________________________________________________________________________________________ shared_res_23_3rd (Conv1D) (None, 7, 1024) 1049600 leaky_re_lu[33][0] __________________________________________________________________________________________________ input_3 (InputLayer) [(None, 2)] 0 __________________________________________________________________________________________________ shared_res_23_3rd_batchnorm (Ba (None, 7, 1024) 4096 shared_res_23_3rd[0][0] __________________________________________________________________________________________________ flatten_3 (Flatten) (None, 2) 0 input_3[0][0] __________________________________________________________________________________________________ add_9 (Add) (None, 7, 1024) 0 shared_res_23_3rd_batchnorm[0][0] max_pooling1d_8[0][0] __________________________________________________________________________________________________ r2_inz_dense (Dense) (None, 3072) 9216 flatten_3[0][0] __________________________________________________________________________________________________ tf.reshape_1 (TFOpLambda) (None, 3, 1024, 1) 0 r2_inz_dense[0][0] __________________________________________________________________________________________________ max_pooling1d_9 (MaxPooling1D) (None, 3, 1024) 0 leaky_re_lu[34][0] __________________________________________________________________________________________________ batch_normalization_1 (BatchNor (None, 3, 1024, 1) 4 tf.reshape_1[0][0] __________________________________________________________________________________________________ tf.expand_dims_1 (TFOpLambda) (None, 3, 1024, 1) 0 max_pooling1d_9[0][0] __________________________________________________________________________________________________ concatenate_3 (Concatenate) (None, 3, 1024, 2) 0 tf.expand_dims_1[0][0] leaky_re_lu[44][0] __________________________________________________________________________________________________ flatten_4 (Flatten) (None, 6144) 0 concatenate_3[0][0] __________________________________________________________________________________________________ r2_dense_1 (Dense) (None, 1024) 6292480 flatten_4[0][0] __________________________________________________________________________________________________ r2_dense_1_batchnorm (BatchNorm (None, 1024) 4096 r2_dense_1[0][0] __________________________________________________________________________________________________ r2_dense_2 (Dense) (None, 1024) 1049600 leaky_re_lu[45][0] __________________________________________________________________________________________________ r2_dense_2_batchnorm (BatchNorm (None, 1024) 4096 r2_dense_2[0][0] __________________________________________________________________________________________________ r2_dense_3 (Dense) (None, 1024) 1049600 leaky_re_lu[46][0] __________________________________________________________________________________________________ r2_dense_3_batchnorm (BatchNorm (None, 1024) 4096 r2_dense_3[0][0] __________________________________________________________________________________________________ r2_dense_4 (Dense) (None, 1024) 1049600 leaky_re_lu[47][0] __________________________________________________________________________________________________ r2_dense_4_batchnorm (BatchNorm (None, 1024) 4096 r2_dense_4[0][0] __________________________________________________________________________________________________ JointVonMisesFisher_mean (Dense (None, 3) 3075 leaky_re_lu[48][0] __________________________________________________________________________________________________ JointVonMisesFisher_logvar (Den (None, 1) 1025 leaky_re_lu[48][0] __________________________________________________________________________________________________ concatenate_4 (Concatenate) (None, 4) 0 JointVonMisesFisher_mean[0][0] JointVonMisesFisher_logvar[0][0] ================================================================================================== Total params: 21,775,368 Trainable params: 21,739,398 Non-trainable params: 35,970 __________________________________________________________________________________________________