maxtext
maxtext copied to clipboard
profiler to start only for the last step
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.