Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

About pretrain model #22

Open
xu19971109 opened this issue Nov 5, 2024 · 0 comments
Open

About pretrain model #22

xu19971109 opened this issue Nov 5, 2024 · 0 comments

Comments

@xu19971109
Copy link

DVIS-DAQ is an impressive work. I have a question regarding the training strategy:

The paper mentions that the segmenter, tracker, and temporal refiner are trained in separate stages with frozen components. From the code, I see that MinVIS, Online, and Offline modes can be used to train these three components individually. However, I couldn’t find where to configure the corresponding weights for each stage.

Should I load the weights from the previous stage in the configuration file like this?

MODEL:
  WEIGHTS: 'MinVIS.pth'

MODEL:
  WEIGHTS: 'Online.pth'

Additionally, I noticed that you have shared pre-trained weights on Hugging Face. Could you let me know which one corresponds to the COCO & YTVIS-2021 VIT finetuned segmenter?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant