def get_defaut_config(): c = Config() c.INPUT_TYPE = "blind" c.MODEL_PATH = "data/checkpoints/blind.pth" c.RL.PPO.hidden_size = 512 c.RANDOM_SEED = 7 c.TORCH_GPU_ID = 0 return c
def get_defaut_config(): c = Config() c.INPUT_TYPE = "blind" c.MODEL_PATH = "data/checkpoints/blind.pth" c.RESOLUTION = 256 c.HIDDEN_SIZE = 512 c.RANDOM_SEED = 7 c.PTH_GPU_ID = 0 return c