-
Notifications
You must be signed in to change notification settings - Fork 2
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
about the train code, train_base.py of BAM is not used to this repo's src/model/pspnet.py #3
Comments
Dear Sina, |
Hi, I used the same
You can paste it somewhere after the BAM model is loaded, for example after this line, run the code until the end of this snippet (maybe add an Let me know if this doesn't resolve the issue. |
Thanks for the answer! It solved perfectly the problem I was having. In principle, the code you provided loads the original pspnet model parameters into the structure corresponding to the current repo's pspnet. Very ingenious. Thank you again! |
No worries, I'm happy it helped! |
@Nevaeh7 are you available for a quick talk please |
Hello, I would like to ask if I can use a network other than PSPNet, such as SegFormer from the MMSEG framework, since you mentioned in the readme that any model trained on the base class can be used. If so, how can I modify the code in your repository? |
Hi, |
No description provided.
The text was updated successfully, but these errors were encountered: