(self, session_class)
| 89 | return self.init_group.used_nodes() |
| 90 | |
| 91 | def compile(self, session_class): |
| 92 | self._nodes_to_checkpoint = self.nodes_to_checkpoint() |
| 93 | self.init_group = session_class.compile(self.init_group) |
| 94 | self.epoch_group = session_class.compile(self.epoch_group) |
| 95 | self.download_group = session_class.compile(self.download_group) |
| 96 | self.exit_group = session_class.compile(self.exit_group) |
| 97 | |
| 98 | def __enter__(self): |
| 99 | super().__enter__() |