(self)
| 68 | self._inject() |
| 69 | |
| 70 | def _inject(self): |
| 71 | trainer = self.trainer # noqa |
| 72 | if self._shell == 'ipython': |
| 73 | import IPython as IP # noqa |
| 74 | IP.embed() |
| 75 | elif self._shell == 'pdb': |
| 76 | import pdb # noqa |
| 77 | pdb.set_trace() |
| 78 | |
| 79 | def _after_train(self): |
| 80 | if os.path.isfile(self._file): |