a0ce9c776719da042d58022a31db5e99755588ad,kfac/python/ops/kfac_utils/periodic_inv_cov_update_kfac_opt.py,PeriodicInvCovUpdateKfacOpt,kfac_update_ops,#PeriodicInvCovUpdateKfacOpt#,93

Before Change

    Returns:
      An op that when run will run the update ops at their update frequencies.
    
    with tf.variable_scope(self.get_name()):
      (cov_update_thunks,
       inv_update_thunks) = self.make_vars_and_create_op_thunks()
      counter = self.counter
      prev_counter = tf.assign(
          tf.get_variable(
              "prev_counter", dtype=tf.int64, shape=(), trainable=False,
              initializer=tf.zeros_initializer, use_resource=True),

After Change

      (cov_update_thunks,
       inv_update_thunks) = self.make_vars_and_create_op_thunks()
    else:
      (_, cov_update_thunks,
       _, inv_update_thunks) = self.create_ops_and_vars_thunks()

    should_do_cov_updates = tf.equal(tf.mod(self.counter,
                                            self._cov_update_every), 0)

Instances