training / flax /run_speed_pt.py

Commit History

Saving train state of step 50
55f3766
verified

supawichwac commited on