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

facebookresearch / swav

PyTorch implementation of SwAV https//arxiv.org/abs/2006.09882

Other

2.01k stars 280 forks source link

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

Open xgyyao opened 3 years ago

xgyyao commented 3 years ago

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?