Commit ce5b87d

committed

Fix eod masking + strategy refactoring

1 parent 9e786c8 commit ce5b87dCopy full SHA for ce5b87d

4 files changed

out.txt
src/maxtext/trainers/post_train/distillation
- distillation_utils.py
- train_distill.py
tests/post_training/unit
- train_distill_test.py

Comments

(0)

Provide feedback