working on squad

2018-11-02 04:07:52 +01:00
parent e61db0d1c0
commit 8e81e5e6ff
2 changed files with 64 additions and 68 deletions
--- a/run_classifier_pytorch.py
+++ b/run_classifier_pytorch.py
@@ -440,7 +440,7 @@ def main():
    else:
        device = torch.device("cuda", args.local_rank)
        n_gpu = 1
-        print("Initializing the distributed backend: NCCL")
+        # print("Initializing the distributed backend: NCCL")
    print("device", device, "n_gpu", n_gpu)

    if not args.do_train and not args.do_eval: