search

google-research / t5x

Apache License 2.0
2.58k stars 297 forks source link

Propagate use_orbax_format to _get_optimizer_state_dict. #1506

Closed copybara-service[bot] closed 4 months ago

copybara-service[bot] commented 5 months ago

Propagate use_orbax_format to _get_optimizer_state_dict.