google-deepmind / kfac-jax

Second Order Optimization and Curvature Estimation with K-FAC in JAX.
Apache License 2.0
250 stars 23 forks source link

Changing optimizer to throw an exception when using burnin without a provided data iterator instead of silently skipping burnin. #277

Closed copybara-service[bot] closed 1 month ago

copybara-service[bot] commented 1 month ago

Changing optimizer to throw an exception when using burnin without a provided data iterator instead of silently skipping burnin.