search

Justin-Tan / high-fidelity-generative-compression

Pytorch implementation of High-Fidelity Generative Image Compression + Routines for neural image compression
Apache License 2.0
411 stars 77 forks source link

ValueError: loaded state dict contains a parameter group that doesn't match the size of optimizer's group #27

Open Hdusk opened 3 years ago

Hdusk commented 3 years ago

It happened when I run 'python3 train.py --model_type compression_gan --regime med --warmstart -ckpt path/to/trained/model --likelihood_type logistic' with 'HIFIC-med'

Justin-Tan commented 3 years ago

Yeah I think somewhere along the line I modified the models to make them more efficient and this got out of sync with the pretrained models - so you'll have to train them yourself unfortunately.