Rayhane-mamah / Efficient-VDVAE

Official Pytorch and JAX implementation of "Efficient-VDVAE: Less is more"
https://arxiv.org/abs/2203.13751
MIT License
188 stars 21 forks source link

som questions about checkpoint #11

Closed miaoYuanyuan closed 1 year ago

miaoYuanyuan commented 1 year ago

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