(self, state_dict, strict=True)
| 86 | return new_state_dict |
| 87 | |
| 88 | def load_state_dict(self, state_dict, strict=True): |
| 89 | new_state_dict = self.preprocess_state_dict(state_dict) |
| 90 | super().load_state_dict(new_state_dict, strict) |
| 91 | |
| 92 | def step_log_dict(self): |
| 93 | return { |
no test coverage detected