RepViT icon indicating copy to clipboard operation
RepViT copied to clipboard

repvit-sam train self built datasets and an error

Open rememberBr opened this issue 2 years ago • 2 comments

I have two questions:

  1. Will repvit-sam provide a pipeline for training self built datasets? Make it only segment the categories we need, not anything else
  2. 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

rememberBr avatar Dec 14 '23 01:12 rememberBr