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

How to config use_layers for eff or resnet #12

Open
FRAOTIAC opened this issue Apr 11, 2022 · 2 comments
Open

How to config use_layers for eff or resnet #12

FRAOTIAC opened this issue Apr 11, 2022 · 2 comments

Comments

@FRAOTIAC
Copy link

Hi guys, great work!
I wanna try eff_b7 or resnet50 as backbone, but seems the code is default to swin.That if simply change backbone to eff_b7 or resnet, the assert inside model defination will raise error:

        assert len(use_layers) == self.num_layers
        assert len(use_selections) == len(use_layers)

I wonder what is the correct use_layers setting for such backbone?

@FRAOTIAC
Copy link
Author

Update:
I managed to get the code running with eff_b7 backbone with the settings below:

        args.use_layers = [False, False, False, False, False, True, True, True, True]
        args.use_selections = [False, False, False, False, False, True, True, True, True]
        args.num_selects = [0,0,0,0,0,2048, 512, 128, 32]
        args.use_gcn = False

Are these settings the same as your base settings? Thank you in advance for your help.

@chou141253
Copy link
Owner

Yes, and we release a new version (Hope that helps).

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