Skip to content

Commit eeaccb4

Browse files
committed
Revert "Revert "Drop the initialization layer""
This reverts commit d6a73ef.
1 parent 05c7acb commit eeaccb4

File tree

1 file changed

+7
-11
lines changed

1 file changed

+7
-11
lines changed

networks.py

Lines changed: 7 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -476,17 +476,13 @@ def thick_conv_lstm_readout_net(feature_map_in_seqs, feature_map_size, drop_rate
476476
temp_shape = [int(s) for s in temp_shape]
477477
x = tf.reshape(x, [batch_size, n_step, temp_shape[0], temp_shape[1], temp_shape[2]])
478478

479-
initial_c = layers.Conv2D(5, (1, 1), activation='tanh')(tf.reduce_mean(x, axis=1))
480-
initial_h = layers.Conv2D(5, (1, 1), activation='tanh')(tf.reduce_mean(x, axis=1))
481-
482-
conv_lstm = layers.ConvLSTM2D(filters=5,
483-
kernel_size=(3,3),
484-
strides=(1,1),
485-
padding='same',
486-
dropout=drop_rate,
487-
recurrent_dropout=drop_rate,
488-
return_sequences=True)
489-
x = conv_lstm([x, initial_c, initial_h])
479+
x = layers.ConvLSTM2D(filters=5,
480+
kernel_size=(3,3),
481+
strides=(1,1),
482+
padding='same',
483+
dropout=drop_rate,
484+
recurrent_dropout=drop_rate,
485+
return_sequences=True)(x)
490486

491487
x = wps.TimeDistributed(layers.Conv2D(1, (1, 1), activation='linear'))(x)
492488

0 commit comments

Comments
 (0)