RepViT
RepViT copied to clipboard
repvit-sam train self built datasets and an error
I have two questions:
- Will repvit-sam provide a pipeline for training self built datasets? Make it only segment the categories we need, not anything else
- I used tim=0.9.12 and run: Python app\app.py, on the 117 line of timm\models_factory.py will throw an error: TypeError: revit() got an unexpected keyword argument 'pretrained'_ Cfg ' timm\models_factory.py Line 117: Model=create_ Fn( Pretrained=pretrained, Pretrained_Cfg=pretrained_Cfg, Pretrained_Cfg_Overlay=pretrained_Cfg_Overlay, **Kwargs, ) if remove Pretrained_Cfg_Overlay and Pretrained_Cfg, I can successfully run it