support wandb sweep

https://github.com/wandb/examples/tree/master/examples/pytorch/pytorch-cnn-fashion

Author: TITC

pix2tex/train.py

@@ -135,4 +135,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)