added back exit

Author: amitsrivastava78

opt.py

@@ -459,8 +459,8 @@ def sync():
         if args.benchmark:
             input_ids = next(iter(dataloader))[0][:, :args.benchmark]
             benchmark(model, input_ids, check=args.check)
-    # if args.load:
-    #     exit()
+    if args.load:
+        exit()
 
     datasets = ['wikitext2', 'ptb'] 
     if args.new_eval: