182:        _rs = _os.environ.get("PHASE2_RESUME_STEP")
183-        if _rs:
184-            self.steps = int(_rs)
185-            if self.accelerator.is_main_process:
186:                print(f"[PHASE2_RESUME_STEP] self.steps={self.steps} "
187-                      f"(LR schedule positioned; fresh optimizer; no data skip)")
188-
189-        self.num_epoch = num_epoch
190-        self.save_every = save_every
191-        self.sample_every = sample_every
192-        self.fid_every = fid_every
193-        self.max_grad_norm = max_grad_norm
194-
195-        self.cfg = cfg
196-        self.test_num_slots = test_num_slots
197-        if self.test_num_slots is not None:
198-            self.test_num_slots = min(self.test_num_slots, self.num_slots)
199-        else:
200-            self.test_num_slots = self.num_slots