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

question about training process #5

Open
ccw12345 opened this issue Sep 13, 2021 · 1 comment
Open

question about training process #5

ccw12345 opened this issue Sep 13, 2021 · 1 comment

Comments

@ccw12345
Copy link

I am confused about your training process. you load the pretrained model,i want to know the pretrained model and other parts of the model are independently trained or freeze the pretrained model,only train the other parts of the model?

@wzjialang
Copy link
Owner

In our paper, we use SE-ResNet50 as our backbone and initialize it with weights pretrained on ImageNet (available at https://github.com/moskomule/senet.pytorch/releases/download/archive/seresnet50-60a8950a85b2b.pkl). After that, we trained our whole model end-to-end by using the remote sensing change detection dataset.

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

2 participants