pub_soft/foldingdiff
3
0
Fork 0
mirror of https://github.com/microsoft/foldingdiff.git synced 2026-06-04 21:34:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

Support for running without a config json

Browse Source
This commit is contained in:
Kevin Wu
2022-07-25 19:10:38 +00:00
parent b435ae587c
commit 31cc02c2ce
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
bin/train.py
View File

@@ -153,7 +153,7 @@ def build_parser() -> argparse.ArgumentParser:
parser = argparse.ArgumentParser(
usage=__doc__, formatter_class=argparse.ArgumentDefaultsHelpFormatter,
)
parser.add_argument("config_json", type=str, help="json of params")
parser.add_argument("-c", "--config", type=str, help="json of params")
parser.add_argument(
"-o",
"--outdir",
@@ -173,8 +173,10 @@ def main():
args = parser.parse_args()
# Load in parameters and run training loop
with open(args.config_json) as source:
config_args = json.load(source)
config_args = {} # Empty dictionary as default
if args.config:
with open(args.config) as source:
config_args = json.load(source)
train(results_dir=args.outdir, toy=args.toy, **config_args)