(self, destination=None, prefix='', keep_vars=False)
| 93 | return self.module(*inputs, **kwargs) |
| 94 | |
| 95 | def state_dict(self, destination=None, prefix='', keep_vars=False): |
| 96 | # [h.remove() for h in self.hook_handles] |
| 97 | sd = self.module.state_dict(destination, prefix, keep_vars) |
| 98 | return sd |
| 99 | |
| 100 | def load_state_dict(self, state_dict, strict=True): |
| 101 | return self.module.load_state_dict(state_dict, strict=strict) |