AICIty-reID-2020
AICIty-reID-2020 copied to clipboard
--init_name pretrained model
@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