ae585178b8f190bcbf7f1433a731fa27fc32271a,niftynet/engine/application_driver.py,ApplicationDriver,_create_graph,#ApplicationDriver#,88

Before Change

            self.app.set_output_op(net_outputs)
            variable_averages = tf.train.ExponentialMovingAverage(0.9)
            self._moving_ave_op = variable_averages.apply(
                tf.trainable_variables())

            if self.is_training:
                averaged_grads = util.average_gradients(training_grads)
                self.app.set_network_update_op(averaged_grads)

After Change

            // moving average operation
            variable_averages = tf.train.ExponentialMovingAverage(0.9)
            trainables = tf.trainable_variables()
            moving_ave_op = variable_averages.apply(trainables)

            // training operation
            if self.is_training:
                updates_op = [moving_ave_op]

Instances