Load pretrain state
(self, path: str)
| 207 | pickle.dump(state, open(self.output_path / 'history.pkl', 'wb')) |
| 208 | |
| 209 | def load_state(self, path: str): |
| 210 | """ |
| 211 | Load pretrain state |
| 212 | """ |
| 213 | path = Path(path) |
| 214 | if (path / 'dataset.csv').is_file(): |
| 215 | self.dataset.load_dataset(path / 'dataset.csv') |
| 216 | if (path / 'history.pkl').is_file(): |
| 217 | state = pickle.load(open(path / 'history.pkl', 'rb')) |
| 218 | self.eval.history = state['history'] |
| 219 | self.batch_id = state['batch_id'] |
| 220 | self.cur_prompt = state['prompt'] |
| 221 | self.task_description = state['task_description'] |
| 222 | self.patient = state['patient'] |
| 223 | |
| 224 | def step(self): |
| 225 | """ |
no test coverage detected