(self)
| 226 | param_group["lr"] = lr |
| 227 | |
| 228 | def log_step(self): |
| 229 | logger.logkv("step", self.step + self.resume_step) |
| 230 | logger.logkv("samples", (self.step + self.resume_step + 1) * self.global_batch) |
| 231 | |
| 232 | def save(self): |
| 233 | def save_checkpoint(rate, params): |