Extend error reporting with mast error parsing format, add @record de…

…corator and info log level to launch

Summary: Extend error reporting with mast error parsing format, add record decorator and info log level to launch

Reviewed By: chrisxcai

Differential Revision: D29783466

fbshipit-source-id: 626679c58158f613211ab8beb174aa8e4f40411a

torchelastic/distributed/launch.py

@@ -8,14 +8,16 @@
 
 import os
 
-from torch.distributed.run import parse_args, run
+os.environ["LOGLEVEL"] = "INFO"
+
+# Since logger initialized during imoprt statement
+# the log level should be set first
+from torch.distributed.run import main as run_main
 
 
 def main(args=None):
-    args = parse_args(args)
-    run(args)
+    run_main(args)
 
 
 if __name__ == "__main__":
-    os.environ["LOGLEVEL"] = "INFO"
     main()