(self)
| 83 | self._nodes_to_checkpoint = nodes_to_checkpoint |
| 84 | |
| 85 | def nodes_to_checkpoint(self): |
| 86 | if self._nodes_to_checkpoint: |
| 87 | return self._nodes_to_checkpoint |
| 88 | else: |
| 89 | return self.init_group.used_nodes() |
| 90 | |
| 91 | def compile(self, session_class): |
| 92 | self._nodes_to_checkpoint = self.nodes_to_checkpoint() |