Load the state dictionary.
(
self,
state_dict: T.Dict[str, T.Any],
)
| 2455 | return to_save |
| 2456 | |
| 2457 | def load_state_dict( |
| 2458 | self, |
| 2459 | state_dict: T.Dict[str, T.Any], |
| 2460 | ): |
| 2461 | """Load the state dictionary.""" |
| 2462 | for name in ['rgb', 'depth', 'normal_w', 'hit_map', 'feature']: |
| 2463 | setattr(self, name, state_dict.get(name, None)) |
| 2464 | self.camera.load_state_dict(state_dict.get('camera', None)) |
| 2465 | |
| 2466 | def save( |
| 2467 | self, |
nothing calls this directly
no test coverage detected