Closed copybara-service[bot] closed 1 month ago
Add support for passing a dataset to flax_e2e_model.train_and_evaluate and make statistics printing easier to read.
flax_e2e_model.train_and_evaluate
Reusing the dataset saves a significant amount of time when comparing multiple configurations.
Add support for passing a dataset to
flax_e2e_model.train_and_evaluateand make statistics printing easier to read.Reusing the dataset saves a significant amount of time when comparing multiple configurations.