Closed jaideep11061982 closed 3 years ago
@jaideep11061982 use the model factory or at least copy what it's doing, pretrained_backbone is to load the pretrained backbone weights, not the pretrained dection model weights
https://github.com/rwightman/efficientdet-pytorch/blob/master/effdet/factory.py#L17-L54
sure thanks i figured it out.
I use this . it works
net=create_model_from_config(config,pretrained=True,bench_task='train',bench_labeler=True)
sure thanks i figured it out. I use this . it works
net=create_model_from_config(config,pretrained=True,bench_task='train',bench_labeler=True)
What are the inputs
Describe the bug A clear and concise description of what the bug is. When i call the Effdet class with pretrained_backbone=True , code is loading the pretrained weights of corresponding efficientnet imagenet model but not effdet weights.Below is message i get for pretrained =True
Downloading: "https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/tf_efficientnet_b4_aa-818f208c.pth" to /root/.cache/torch/hub/checkpoints/tf_efficientnet_b4_aa-818f208c.pth
1. 2.
Expected behavior A clear and concise description of what you expected to happen. it should load the weights as per the url given in https://github.com/rwightman/efficientdet-pytorch/blob/abba1d5a3611471ac88d49a473f993f72f9e1aba/effdet/config/model_config.py Screenshots If applicable, add screenshots to help explain your problem.
Desktop (please complete the following information):
conda list
, 1.7.0 py3.8_cuda11.0.221_cudnn8.0.3_0]Additional context Add any other context about the problem here.