File size: 842 Bytes
f065e53 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | 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
|