diff --git a/configs/iccv/config_abl_pyr_32_64_128_pre_cape_f.py b/configs/iccv/config_abl_pyr_32_64_128_pre_cape_f.py index b5a0b7c..d831507 100644 --- a/configs/iccv/config_abl_pyr_32_64_128_pre_cape_f.py +++ b/configs/iccv/config_abl_pyr_32_64_128_pre_cape_f.py @@ -22,7 +22,7 @@ config['use_offsets'] = False config['use_norm_loss'] = True config['use_curvature'] = True -config['pretrained_model'] = 'checkpoints/iccv/iccv_pyr_32_64_128_pre_cape_f' +config['pretrained_model'] = 'checkpoints/iccv/iccv_pyr_32_64_128' config['pretrained_model_iteration'] = '' config['pyramid'] = dict()