pre-train weight for mask_generator.py

[chinglun]

When i run mask_generator.py ,it need some pre-train weights like /users/gyungin/sos/networks/pretrained/moco_v2_800ep_pretrain.pth.tar
Is it necessary? If so, how can i get it?

Thanks for your response!

[noelshin]

Dear @baron0830

Apologies for the late reply and thanks for your interest in our work.
Yes, they are necessary and you can download here:

For MoCov2: https://dl.fbaipublicfiles.com/moco/moco_checkpoints/moco_v2_800ep/moco_v2_800ep_pretrain.pth.tar
For SwAV: https://dl.fbaipublicfiles.com/deepcluster/swav_800ep_pretrain.pth.tar

I believe in case of DINO, it'll download directly from a given URL in the code.

Hope this helps,
Gyungin