repvit-sam train self built datasets and an error

[rememberBr]

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

[jameslahm]

Thanks for your interest.

Will repvit-sam provide a pipeline for training self built datasets? Make it only segment the categories we need, not anything else

We will release the code for training self built datasets.

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==0.5.4 should be fine.

[rememberBr]

Looking forward to your work and hoping to see further updates as soon as possible