maxtext icon indicating copy to clipboard operation
maxtext copied to clipboard

profiler to start only for the last step

Open ultrons opened this issue 2 years ago • 0 comments

Starting at step 0 results in an error if training is re-run from the checkpoint saved at step 0 or later. This change starts the profile at the end of penultimate step.

ultrons avatar Oct 06 '23 23:10 ultrons