Load the state dictionary.
(
self,
state_dict: T.Dict[str, T.Any],
)
| 1127 | return to_save |
| 1128 | |
| 1129 | def load_state_dict( |
| 1130 | self, |
| 1131 | state_dict: T.Dict[str, T.Any], |
| 1132 | ): |
| 1133 | """Load the state dictionary.""" |
| 1134 | for name in ['origins_w', 'directions_w']: |
| 1135 | setattr(self, name, state_dict.get(name, None)) |
| 1136 | |
| 1137 | def save( |
| 1138 | self, |