(self, cp)
| 55 | if disc_unexpected: print(f"Unexpected keys in netBase:\n{disc_unexpected}") |
| 56 | |
| 57 | def load_optimizer_state(self, cp): |
| 58 | super().load_optimizer_state(cp) |
| 59 | self.optimizerDisc.load_state_dict(cp["optimizerDisc"]) |
| 60 | |
| 61 | def get_model_state(self): |
| 62 | state = super().get_model_state() |