Merge branch 'data-parallelism' of https://github.com/TITC/LaTeX-OCR into pr/150

Author: lukas-blecher

pix2tex/train.py

@@ -112,4 +112,5 @@ def save_models(e, step=0):
         if not parsed_args.resume:
             args.id = wandb.util.generate_id()
         wandb.init(config=dict(args), resume='allow', name=args.name, id=args.id)
+        args = Munch(wandb.config)
     train(args)