facebookresearch/swav

Different weight size between pretrained model and the model trained on my own datasets of deepclusterv2

xgyyao opened this issue · 0 comments

Thank for your great project!
The size of deepclusterv2_800ep_pretrain.pth.tar is 112M,but the size of the model trained on my own datasets by main_deepclusterv2.py is quite different:
size of ckp-99.pth is 219M
size of checkpoint.pth.tar is 219M

How can I get my own model the same size as deepclusterv2_800ep_pretrain.pth.tar?