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

Is this repo really needed for RESNET18 ? #17

Open
ExtReMLapin opened this issue Mar 11, 2022 · 2 comments
Open

Is this repo really needed for RESNET18 ? #17

ExtReMLapin opened this issue Mar 11, 2022 · 2 comments

Comments

@ExtReMLapin
Copy link

Hello,

I recently started using detectron2, if i want to train a custom dataset on resnet18, don't I just need to take RESNET50 model, set DEPTH to 18 and set RES2_OUT_CHANNELS to 64 ?

This way :

        data["MODEL"]["RESNETS"]["DEPTH"] = 18
        data["MODEL"]["RESNETS"]["RES2_OUT_CHANNELS"] = 64

And obviously, without any trained model to import weights from, set weights to "" and unfreeze first two layers

        cfg.MODEL.WEIGHTS = ""
        cfg.MODEL.BACKBONE.FREEZE_AT = 0

As only for RESNET18, what does this repository brings compared to what I just showed above ?

Apologies if it's a naive question, I'm new to this whole deeplearning stuff.

Thanks

@Huxwell
Copy link

Huxwell commented Sep 15, 2023

I know a long time has passed, but I was able to use resnet18/34 successfully with your approach, in fact thanks for the idea!
That gives me some speedup, maybe x1.3-1.5.
Although it has to be trained from scratch, because I didn't find any compatible COCO pretrained weights online.
Also, the DefaultPredictor should be avoided when running inference with your approach.

@ExtReMLapin
Copy link
Author

Dang yeah mate, it's been some time

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