small bugfix regarding saving the step

Author: alex-haas

rin_pytorch/rin_pytorch.py

@@ -870,7 +870,7 @@ def save(self, milestone):
             return
 
         data = {
-            'step': self.step,
+            'step': self.step + 1,
             'model': self.accelerator.get_state_dict(self.model),
             'opt': self.opt.state_dict(),
             'ema': self.ema.state_dict(),