Returns a dictionary that can be saved or load.
(self)
| 1120 | return self.origins_w.device |
| 1121 | |
| 1122 | def state_dict(self) -> T.Dict[str, T.Any]: |
| 1123 | """Returns a dictionary that can be saved or load.""" |
| 1124 | to_save = dict() |
| 1125 | for name in ['origins_w', 'directions_w']: |
| 1126 | to_save[name] = getattr(self, name, None) |
| 1127 | return to_save |
| 1128 | |
| 1129 | def load_state_dict( |
| 1130 | self, |
no outgoing calls
no test coverage detected