light weight backbone

[ForestWang]

would you support some light weight backbones such as RepVGG which is GPU friendly?

[xoiga123]

Sure thing! We hid it inside backbone.py but now we're bringing it to light (command-line). Use timm to replace your backbone. Try

python train.py -p bdd100k -bb repvgg_b0

This is basically what we do under the hood:

if backbone_name:
            self.encoder = timm.create_model(backbone_name, pretrained=True, features_only=True, out_indices=(2,3,4))  # P3,P4,P5

[MartinPR307]

Hello, after switching to repvgg_b0, the following error occurs：

[MartinPR307]

@datvuthanh @xoiga123

[MartinPR307]

@ForestWang Have you encountered the same problem?

[HT24TH]

I have the same problem @MartinPR307, any suggestion to solve that @datvuthanh @xoiga123. Thank you.