ckpt_dir = 'ckpts/'