search

IDEA-Research / detrex-storage

Apache License 2.0
2 stars 0 forks source link

About weight files #1

Closed wangzhaoyang-508 closed 1 year ago

wangzhaoyang-508 commented 1 year ago

Why does the dino_r50_4scale_12ep.pth only 182MB，but the dino_r50_5scale_12ep.pth is much bigger than that？

rentainhe commented 1 year ago

Why does the dino_r50_4scale_12ep.pth only 182MB，but the dino_r50_5scale_12ep.pth is much bigger than that？

Sry for missing this issue before, this is because we forgot to move the optimizer state in some checkpoints which may increase the weight size

rentainhe commented 1 year ago

Why does the dino_r50_4scale_12ep.pth only 182MB，but the dino_r50_5scale_12ep.pth is much bigger than that？

Sry for missing this issue before, this is because we forgot to move the optimizer state in some checkpoints which may increase the weight size

for some larger model as Swin-Large-DINO, we will remove the optimizer state to make it smaller enough for the git storage

rentainhe commented 1 year ago

I'm closing this issue~ feel free to reopen it if needed