search

hijune6 / Hetero-center-triplet-loss-for-VT-Re-ID

MIT License
45 stars 5 forks source link

test error #1

Closed huangpan2507 closed 3 years ago

huangpan2507 commented 3 years ago

When I run python test_mine_pcb.py --dataset sysu --mode all --pcb on --share_net 2 --w_center 2 --gpu 0 --resume sysu_c_tri_pcb_on_w_tri_2.0_s6_f256_share_net2_base_gm10_k4_p8_lr_0.1_seed_0_best.t It read that :

==> Building model.. ==> Loading data.. ==> Resuming from checkpoint.. ==> no checkpoint found at sysu_c_tri_pcb_on_w_tri_2.0_s6_f256_share_net2_base_gm10_k4_p8_lr_0.1_seed_0_best.t

huangpan2507 commented 3 years ago

When I run python test_mine_pcb.py --dataset sysu --mode all --pcb on --share_net 2 --w_center 2 --gpu 0 --resume sysu_c_tri_pcb_on_w_tri_2.0_s6_f256_share_net2_base_gm10_k4_p8_lr_0.1_seed_0_best.t It read that :

==> Building model.. ==> Loading data.. ==> Resuming from checkpoint.. ==> no checkpoint found at sysu_c_tri_pcb_on_w_tri_2.0_s6_f256_share_net2_base_gm10_k4_p8_lr_0.1_seed_0_best.t

I find what the hell ! the test file had a default config like below:

if dataset == 'sysu':

print('==> Resuming from checkpoint..')

# model_path = checkpoint_path + args.resume
model_path = checkpoint_path + ***sysu_c_tri_pcb_on_w_tri_1.0_s6_f256_share_net2_base_gm_k8_p6_lr_0.1_seed_0_best.t***
if os.path.isfile(model_path):

So modified it to

model_path = checkpoint_path + args.resume