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

…into pr/150
wesleylamlam · May 20, 2022 · de8f7f0 · de8f7f0
2 parents bd66642 + 694d31b
commit de8f7f0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pix2tex/train.py b/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)