(self, items, keep_every_n=None)
| 104 | """Class to manage save and restore for the model and optimizer.""" |
| 105 | |
| 106 | def __init__(self, items, keep_every_n=None): |
| 107 | assert type(items) == dict |
| 108 | assert "model" in items |
| 109 | self._items = items |
| 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"]): |