예제 #1
0
def setup(args):
    if args.config_file.endswith(".yaml"):
        cfg = get_cfg()
        cfg.merge_from_file(args.config_file)
        cfg.SOLVER.BASE_LR = 0.001  # Avoid NaNs. Not useful in this script anyway.
        cfg.merge_from_list(args.opts)
        cfg.freeze()
    else:
        cfg = LazyConfig.load(args.config_file)
        cfg = LazyConfig.apply_overrides(cfg, args.opts)
    setup_logger(distributed_rank=comm.get_rank())
    return cfg
예제 #2
0
def main(args):
    cfg = LazyConfig.load(args.config_file)
    cfg = LazyConfig.apply_overrides(cfg, args.opts)
    default_setup(cfg, args)

    if args.eval_only:
        model = instantiate(cfg.model)
        model = create_ddp_model(model)
        DetectionCheckpointer(model).load(cfg.train.init_checkpoint)
        print(do_test(cfg, model))
    else:
        do_train(args, cfg)
예제 #3
0
def setup(args):
    if args.config_file.endswith(".yaml"):
        cfg = get_cfg()
        cfg.merge_from_file(args.config_file)
        cfg.DATALOADER.NUM_WORKERS = 0
        cfg.merge_from_list(args.opts)
        cfg.freeze()
    else:
        cfg = LazyConfig.load(args.config_file)
        cfg = LazyConfig.apply_overrides(cfg, args.opts)
    setup_logger(name="fvcore")
    setup_logger()
    return cfg
 def test_invalid_overrides(self):
     cfg = LazyConfig.load(self.root_filename)
     with self.assertRaises(KeyError):
         LazyConfig.apply_overrides(cfg, ["lazyobj.x.xxx=123"])
 def test_overrides(self):
     cfg = LazyConfig.load(self.root_filename)
     LazyConfig.apply_overrides(cfg,
                                ["lazyobj.x=123", 'dir1b_dict.a="123"'])
     self.assertEqual(cfg.dir1b_dict.a, "123")
     self.assertEqual(cfg.lazyobj.x, 123)