예제 #1
0
파일: unet_lstm.py 프로젝트: auger1/MMWHS
 def recurrent_cell(self, shape, num_features, postfix, is_training):
     return ConvGRUCell(shape,
                        num_features, [3, 3],
                        activation=tf.nn.relu,
                        data_format=self.data_format,
                        normalize=False,
                        name='gru' + postfix)
예제 #2
0
 def recurrent_cell(self, shape, num_features, postfix, is_training):
     return ConvGRUCell(shape,
                        num_features, [1, 1],
                        activation=tf.nn.relu,
                        data_format=self.data_format,
                        normalization=None,
                        name='gru' + postfix,
                        is_training=is_training,
                        padding=self.padding)