fixed checkpoint_dir path