(self, ckpt: Union[None, str, dict])
| 532 | return batch[self.input_key] |
| 533 | |
| 534 | def apply_ckpt(self, ckpt: Union[None, str, dict]): |
| 535 | if ckpt is None: |
| 536 | return |
| 537 | self.init_from_ckpt(ckpt) |
| 538 | |
| 539 | def init_from_ckpt(self, path, ignore_keys=list()): |
| 540 | sd = torch.load(path, map_location="cpu")["state_dict"] |
nothing calls this directly
no test coverage detected