未验证 提交 41eb09fb 编写于 作者: 农夫三拳_'s avatar 农夫三拳_ 提交者: GitHub

add configuration options (#164)

e.g. python tools/main.py --c configs/stylegan_v2_256_ffhq.yaml -o total_iters=1 log_config.visiual_interval=1
上级 069ee192
...@@ -45,10 +45,10 @@ def parse_args(): ...@@ -45,10 +45,10 @@ def parse_args():
default=False, default=False,
help='skip validation during training') help='skip validation during training')
# config options # config options
parser.add_argument('opts', parser.add_argument("-o",
help='See config for all options', "--opt",
default=None, nargs='+',
nargs=argparse.REMAINDER) help="set configuration options")
#for inference #for inference
parser.add_argument("--source_path", parser.add_argument("--source_path",
......
...@@ -51,6 +51,6 @@ def main(args, cfg): ...@@ -51,6 +51,6 @@ def main(args, cfg):
if __name__ == '__main__': if __name__ == '__main__':
args = parse_args() args = parse_args()
cfg = get_config(args.config_file) cfg = get_config(args.config_file, args.opt)
main(args, cfg) main(args, cfg)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册