| 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 | |