def parse_train_config(config=None):
config = {} if not config else config
c = AttrDict()
c.DATASET_ROOT = config.get("DATASET_ROOT", DATASET_ROOT)
c.JSON_PATH = config.get("JSON_PATH", "train.json")
c.BATCH_SIZE = config.get("BATCH_SIZE", BATCH_SIZE)
c.IMAGE_SIZE = config.get("IMAGE_SIZE", IMAGE_SIZE)
c.WORKERS = config.get("WORKERS", WORKERS)
c.PIN_MEMORY = config.get("PIN_MEMORY", PIN_MEMORY)
c.SHUFFLE = config.get("SHUFFLE", True)
c.LEARNING_RATE = config.get("LEARNING_RATE", LEARNING_RATE)
c.MOMENTUM = config.get("MOMENTUM", MOMENTUM)
c.DAMPENING = config.get("DAMPENING", DAMPENING)
c.BETAS = config.get("BETAS", BETAS)
c.EPS = config.get("EPS", EPS)
c.WEIGHT_DECAY = config.get("WEIGHT_DECAY", WEIGHT_DECAY)
c.MILESTONES = config.get("MILESTONES", MILESTONES)
c.GAMMA = config.get("GAMMA", GAMMA)
c.NUM_EPOCHS = config.get("NUM_EPOCHS", NUM_EPOCHS)
c.TEST = config.get("TEST", TEST)
c.OUT_PATH = config.get("OUT_PATH", OUT_PATH)
c.LOAD_MODEL = config.get("LOAD_MODEL", LOAD_MODEL)
c.SAVE_MODEL = config.get("SAVE_MODEL", SAVE_MODEL)
c.CHECKPOINT_FILE = config.get("CHECKPOINT_FILE", CHECKPOINT_FILE)
return c
def parse_test_config(config=None):
config = {} if not config else config
c = AttrDict()
c.DATASET_ROOT = config.get("DATASET_ROOT", DATASET_ROOT)
c.JSON_PATH = config.get("JSON_PATH", "test.json")
c.BATCH_SIZE = config.get("BATCH_SIZE", BATCH_SIZE)
c.IMAGE_SIZE = config.get("IMAGE_SIZE", IMAGE_SIZE)
c.WORKERS = config.get("WORKERS", WORKERS)
c.PIN_MEMORY = config.get("PIN_MEMORY", PIN_MEMORY)
c.SHUFFLE = config.get("SHUFFLE", False)
c.OUT_PATH = config.get("OUT_PATH", OUT_PATH)
c.LOAD_MODEL = config.get("LOAD_MODEL", True)
c.CHECKPOINT_FILE = config.get("CHECKPOINT_FILE", CHECKPOINT_FILE)
return c