(self, path, ignore_keys=list())
| 57 | # self.register_buffer("mu_sig_bank", torch.empty(0)) |
| 58 | |
| 59 | def init_from_ckpt(self, path, ignore_keys=list()): |
| 60 | sd = torch.load(path, map_location="cpu")["state_dict"] |
| 61 | keys = list(sd.keys()) |
| 62 | for k in keys: |
| 63 | for ik in ignore_keys: |
| 64 | if k.startswith(ik): |
| 65 | print("Deleting key {} from state_dict.".format(k)) |
| 66 | del sd[k] |
| 67 | self.load_state_dict(sd, strict=False) |
| 68 | print(f"Restored from {path}") |
| 69 | |
| 70 | @contextmanager |
| 71 | def ema_scope(self, context=None): |