flax
b27e7846 - Updates references to flax.training.train_state.

Commit

5 years ago

Updates references to flax.training.train_state.

References

#1323 - Renames `train_state` to `state`.

Author

andsteing

andsteing

Committer

andsteing

andsteing

Parents

Loading