AICIty-reID-2020 icon indicating copy to clipboard operation
AICIty-reID-2020 copied to clipboard

--init_name pretrained model

Open vokhidovhusan opened this issue 2 years ago • 0 comments

@layumi Hi. May I ask you something again? I would like to use my own previously trained model as a pre-trained model. Let's say I have SE_imbalance_s1_128_p0.5_lr2_mt_d0_b24+v+aug_20220701 already trained model. And I would like to use my trained model as a pre-trained model for my next training. How could I do this?

I have tried as following: python train_2020.py --data_dir ../../data/reid_data --name SE_imbalance_s1_128_p0.5_lr2_mt_d0_b24+v+aug --warm_epoch 5 --droprate 0 --stride 1 --erasing_p 0.5 --autoaug --inputsize 128 --lr 0.02 --use_SE --gpu_ids 0,1 --train_virtual --batchsize 128 --init_name SE_imbalance_s1_128_p0.5_lr2_mt_d0_b24+v+aug_20220701

however I got following error: ModuleAttributeError("'{}' object has no attribute '{}'".format( torch.nn.modules.module.ModuleAttributeError: 'SENet' object has no attribute 'conv1'

Thanks

vokhidovhusan avatar Jul 08 '22 01:07 vokhidovhusan