(self)
| 153 | h.after_train() |
| 154 | |
| 155 | def before_step(self): |
| 156 | # Maintain the invariant that storage.iter == trainer.iter |
| 157 | # for the entire execution of each step |
| 158 | self.storage.iter = self.iter |
| 159 | |
| 160 | for h in self._hooks: |
| 161 | h.before_step() |
| 162 | |
| 163 | def after_step(self): |
| 164 | for h in self._hooks: |