diff --git a/pytorch_lightning/utils/debugging.py b/pytorch_lightning/utils/debugging.py new file mode 100644 index 00000000..7a4d1445 --- /dev/null +++ b/pytorch_lightning/utils/debugging.py @@ -0,0 +1,15 @@ +import pdb +import sys + +class ForkedPdb(pdb.Pdb): + """A Pdb subclass that may be used + from a forked multiprocessing child + + """ + def interaction(self, *args, **kwargs): + _stdin = sys.stdin + try: + sys.stdin = open('/dev/stdin') + pdb.Pdb.interaction(self, *args, **kwargs) + finally: + sys.stdin = _stdin \ No newline at end of file