Fix restoring steps

run_mlm_flax_stream.py

@@ -691,7 +691,7 @@ if __name__ == "__main__":
     eval_samples = advance_iter_and_group_samples(training_iter, data_args.num_eval_samples, max_seq_length)
 
     steps = tqdm(range(num_train_steps), desc="Training...", position=0)
-    for step in range(saved_step, num_train_steps):
+    for step in range(0, num_train_steps):
         if step < saved_step:
             steps.update(1)
             continue