KeyError: 'hparams' when running tester

[rudolfwilliam]

I trained the full model on Morpho-MNIST using the command

python -m deepscm.experiments.morphomnist.trainer -e SVIExperiment -m ConditionalVISEM --data_dir /path/to/data --default_root_dir /path/to/checkpoints --decoder_type fixed_var.

This worked without issues. However, for running python -m deepscm.experiments.morphomnist.tester -c /path/to/checkpoint/version_0, I get KeyError: 'hparams' in torch.load(checkpoint_path, map_location=torch.device('cpu'))['hparams'].

Indeed, I saw in the debugger that 'hparams' does not exist as dictionary key. However, there exists a key called 'hyper_parameters'. Upon changing that (which already seems wrong), I get an error further down in the line exp_class.load_from_checkpoint(checkpoint_path, pyro_model=model), saying TypeError: __init__() missing 1 required positional argument: 'hparams'.

I checked the code and to me the errors do seem plausible. I would be more surprised if this code would work for anyone. I did use the correct python & package versions.

[minhkhoale]

Do you still want to fix it, I can help.

[rudolfwilliam]

I fixed it myself in the meantime by rewriting the code. Thanks.

[fred7778]

Hello, I am having the same issue with the hparams.

I solved the first part, by replacing the line

hparams = torch.load(checkpoint_path, map_location=torch.device('cpu'))['hparams']

by the line

hparams = torch.load(checkpoint_path, map_location=torch.device('cpu'))['hyper_parameters']

But I am now stuck with the same message as above:

TypeError: init() missing 1 required positional argument: 'hparams'

at the line

experiment = exp_class.load_from_checkpoint(checkpoint_path, pyro_model=model)

in the file morphomnist/tester.py

Any help will be very welcome!
Thank you very much in advance,