def setup(args): cfg = new_config() cfg.merge_from_file(f'configs/{args.config_file}.yaml') cfg.merge_from_list(args.opts) cfg.NAME = args.config_file if args.log_dir: # Override Output dir cfg.OUTPUT_DIR = path.join(args.log_dir, args.config_file) else: cfg.OUTPUT_DIR = path.join(cfg.OUTPUT_BASE_DIR, args.config_file) os.makedirs(cfg.OUTPUT_DIR, exist_ok=True) if args.data_dir: cfg.DATASETS.TRAIN = (args.data_dir, ) return cfg
"--debug", dest='debug', action="store_true", help="whether to attempt to resume from the checkpoint directory", ) args = parser.parse_known_args()[0] MODEL_NAME = args.config_file CHECKPOINT_NAME = args.checkpoint_name THRESHOLD = args.threshold BATCH_SIZE = args.batch_size TRAIN_TYPE = 'test' torch.manual_seed(0) cfg = new_config() if args.debug: cfg.merge_from_file( f'/home/yonk/urban_dl/urban_dl/configs/damage_detection/{MODEL_NAME}.yaml' ) else: cfg.merge_from_file( f'/Midgard/home/pshi/urban_dl/configs/damage_detection/{MODEL_NAME}.yaml' ) cfg.OUTPUT_DIR = path.join(cfg.OUTPUT_BASE_DIR, MODEL_NAME) net = UNet(cfg) print('========== THRESHOLD', cfg.THRESH) # load state dict full_model_path = os.path.join(cfg.OUTPUT_DIR, CHECKPOINT_NAME)
def load_cfg(cfg_file: Path): cfg = new_config() cfg.merge_from_file(str(cfg_file)) return cfg
def load_cfg(configs_dir: Path, cfg_name: str): configs_file = configs_dir / f'{cfg_name}.yaml' cfg = new_config() cfg.merge_from_file(str(configs_file)) return cfg