Restores model and optimizer from given directory. Specify what shoud be restored Returns: Last training step for the model restored.
(self, model_dir, map_location=None,
step=None, item_keys=["model", "optimizer"])
| 110 | self._keep_every_n = keep_every_n |
| 111 | |
| 112 | def restore(self, model_dir, map_location=None, |
| 113 | step=None, item_keys=["model", "optimizer"]): |
| 114 | """Restores model and optimizer from given directory. |
| 115 | Specify what shoud be restored |
| 116 | |
| 117 | Returns: |
| 118 | Last training step for the model restored. |
| 119 | """ |
| 120 | items2restore = { k: self._items[k] for k in item_keys} |
| 121 | last_step = load_checkpoint( |
| 122 | items2restore, model_dir, map_location, step) |
| 123 | return last_step |
| 124 | |
| 125 | def save(self, model_dir, step): |
| 126 | """Saves model and optimizer to given directory. |
no test coverage detected