From 5adc20723bbff4fea628f526b256a60ccd98dbf5 Mon Sep 17 00:00:00 2001 From: thomwolf Date: Fri, 8 Feb 2019 11:03:59 +0100 Subject: [PATCH] add distant debugging --- examples/run_openai_gpt.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/examples/run_openai_gpt.py b/examples/run_openai_gpt.py index 44400702f0..edee22af2b 100644 --- a/examples/run_openai_gpt.py +++ b/examples/run_openai_gpt.py @@ -99,9 +99,21 @@ def main(): parser.add_argument('--weight_decay', type=float, default=0.01) parser.add_argument('--lm_coef', type=float, default=0.5) parser.add_argument('--n_valid', type=int, default=374) + + parser.add_argument('--server_ip', type=str, default='') + parser.add_argument('--server_port', type=str, default='') args = parser.parse_args() print(args) + # Some distant debugging + # See https://code.visualstudio.com/docs/python/debugging#_attach-to-a-local-script + import ptvsd + print(sys.argv) + print("Waiting for debugger attach") + ptvsd.enable_attach(address=(args.server_ip, args.server_port), redirect_output=True) + ptvsd.wait_for_attach() + + random.seed(args.seed) np.random.seed(args.seed) torch.manual_seed(args.seed)