Added ability to specify estimation mode in constructor of KfacOptimizer. ...
Added ability to specify estimation mode in constructor of KfacOptimizer. This functions analogously to the estimation_mode argument to FisherEstimator (and is just passed directly to the FisherEstimator that the optimizer constructs). PiperOrigin-RevId: 175897318
Loading
Please sign in to comment