search

oriondollar / TransVAE

A Transformer Based VAE Architecture for De Novo Molecular Design
MIT License
88 stars 23 forks source link

Can't load checkpoints for sampling #9

Open htalibart opened 1 year ago

htalibart commented 1 year ago

Hi, thank you for making your code public!

when running the command provided in the "Sampling" section of the README

python scripts/sample.py --model transvae --model_ckpt checkpoints/trans4x-256_zinc.ckpt --mols data/zinc_train.txt --sample_mode high_entropy

I get

_pickle.UnpicklingError: invalid load key, 'v'.

any idea how to solve this?

Thanks!