Fail to load checkpoint

[wangyp33]

---

RuntimeError Traceback (most recent call last)
in <cell line: 0>()
2
3 chkpt_dir = '/content/dmae_large_sigma_0.25_mask_0.75_1600e.pth'
----> 4 model_dmae = prepare_model(chkpt_dir, 'dmae_vit_large_patch16')
5 print('Model loaded.')

2 frames
/usr/local/lib/python3.11/dist-packages/torch/serialization.py in init(self, name_or_buffer)
761 class _open_zipfile_reader(_opener):
762 def init(self, name_or_buffer) -> None:
--> 763 super().init(torch._C.PyTorchFileReader(name_or_buffer))
764
765

RuntimeError: PytorchStreamReader failed reading zip archive: failed finding central directory

[quanlin-wu]

I guess it is because you did not download the checkpoint file successfully.