Stage 2 training

[samueleruffino99]

Hello! I would like to train stage 2 only, where can I find the pertained model for stage 1? Apparently on the drive there are online models only for convmae.
Thank you!

[yutaocui]

Hi, there are all models of MixViT/MixViT(ConvMAE) and MixCvT on https://drive.google.com/drive/folders/1C0oECsbTB3uWk2MjrxDuVhIazcT975xb.

[samueleruffino99]

But under MixViT-ConvMAE I can download online models only (already with SPM integrated).

[yutaocui]

You can just drop the SPM module since this module is trained on a separate stage with other weights getting detached.

[samueleruffino99]

Ok, but then at stage 2 SPM only has been trained ? No more training on backbone and box head ?

[yutaocui]

Yes.

[samueleruffino99]

But in this way the SPM would be already pretrained or everything is already implemented in the proper way?

[yutaocui]

We think finetuning the SPM module (excepting for backbone) is enough for filtering reliable samples.