From 2499b0a5fcdb168ccb0095e837b2022953935af2 Mon Sep 17 00:00:00 2001 From: thomwolf Date: Mon, 15 Apr 2019 15:33:04 +0200 Subject: [PATCH] add ptvsd to run_squad --- examples/run_squad.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/examples/run_squad.py b/examples/run_squad.py index bad46203bc..cd85219f5f 100644 --- a/examples/run_squad.py +++ b/examples/run_squad.py @@ -834,7 +834,17 @@ def main(): parser.add_argument('--null_score_diff_threshold', type=float, default=0.0, help="If null_score - best_non_null is greater than the threshold predict null.") + parser.add_argument('--server_ip', type=str, default='', help="Can be used for distant debugging.") + parser.add_argument('--server_port', type=str, default='', help="Can be used for distant debugging.") args = parser.parse_args() + print(args) + + if args.server_ip and args.server_port: + # Distant debugging - see https://code.visualstudio.com/docs/python/debugging#_attach-to-a-local-script + import ptvsd + print("Waiting for debugger attach") + ptvsd.enable_attach(address=(args.server_ip, args.server_port), redirect_output=True) + ptvsd.wait_for_attach() if args.local_rank == -1 or args.no_cuda: device = torch.device("cuda" if torch.cuda.is_available() and not args.no_cuda else "cpu")