File size: 392 Bytes
436b829 | 1 2 3 4 5 6 7 8 9 10 11 12 | import argparse
from omegaconf import OmegaConf
import sys
parser = argparse.ArgumentParser()
parser.add_argument("--cfg_file", default="configs/train_finetune.yaml", type=str)
parser.add_argument("--entry", type=str, default="train_net")
args, unknown = parser.parse_known_args()
cfg = OmegaConf.load(args.cfg_file)
cli_cfg = OmegaConf.from_cli(unknown)
cfg = OmegaConf.merge(cfg, cli_cfg) |