diff --git a/examples/run_openai_gpt.py b/examples/run_openai_gpt.py index 44400702..edee22af 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)