From b41f49dbef6daf826a81ba2f95fee0e6890f252e Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 24 Jul 2019 10:34:21 -0400 Subject: [PATCH] added debugging util --- pytorch_lightning/utils/debugging.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) create mode 100644 pytorch_lightning/utils/debugging.py 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