add speed-up ratio to main.py; fix KeyError in hparams.py

Author: yqzhishen

main.py

@@ -22,6 +22,7 @@
 parser.add_argument('--title', type=str, required=False, help='Title of output file')
 parser.add_argument('--num', type=int, default=1, help='Number of runs')
 parser.add_argument('--seed', type=int, help='Random seed of the inference')
+parser.add_argument('--speedup', type=int, default=0, help='PNDM speed-up ratio')
 args = parser.parse_args()
 
 name = os.path.basename(args.proj).split('.')[0] if not args.title else args.title
@@ -46,6 +47,9 @@
     exp
 ]
 
+if args.speedup > 0:
+    sys.argv += ['--hparams', f'pndm_speedup={args.speedup}']
+
 if not isinstance(params, list):
     params = [params]
 

utils/hparams.py

@@ -89,6 +89,8 @@ def load_config(config_fn):  # deep first
     if args.hparams != "":
         for new_hparam in args.hparams.split(","):
             k, v = new_hparam.split("=")
+            if k not in hparams_:
+                hparams_[k] = eval(v)
             if v in ['True', 'False'] or type(hparams_[k]) == bool:
                 hparams_[k] = eval(v)
             else: