search

rwightman / efficientdet-pytorch

A PyTorch impl of EfficientDet faithful to the original Google impl w/ ported weights
Apache License 2.0
1.58k stars 293 forks source link

what is the difference efficientnet_pytorch and timm.create_model?? #164

Closed SlowMonk closed 3 years ago

SlowMonk commented 3 years ago

try to do the classification with efficnetnet-D0. and found there is two way to do this

    model = timm.create_model('tf_efficientnet_b0_ns', pretrained=True, num_classes=num_classes)

        from efficientnet_pytorch import EfficientNet
        self.efficientnet = EfficientNet.from_pretrained(args.model_name)

the timm model's speed is much faster. what is the difference between thouse two?

rwightman commented 3 years ago

Q for the forums