Rayhane-mamah/Efficient-VDVAE

som questions about checkpoint

Closed this issue · 0 comments

when I download pytorch checkpoint, I have a zip file and unzip it,finnaly I got a directory like
---archive
--data
--data.pkl
--version
--logs-celebAHQ1024_baseline
--hparams-celebahq1024_baseline.cfg

but there is no checkpoint file in it ......

when I download the jax checkpoint , I have a checkpoint file(it has no postfix pth) but when torch.load(xxx), raise error: _pickle.UnpicklingError: unpickling stack underflow