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
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
DEPTHto 18 and setRES2_OUT_CHANNELSto 64 ?This way :
And obviously, without any trained model to import weights from, set weights to
""and unfreeze first two layersAs 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