Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

--init_name pretrained model #51

Open
vokhidovhusan opened this issue Jul 8, 2022 · 0 comments
Open

--init_name pretrained model #51

vokhidovhusan opened this issue Jul 8, 2022 · 0 comments

Comments

@vokhidovhusan
Copy link

vokhidovhusan commented Jul 8, 2022

@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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant