KeyError due to mismatched key name in saved model in dreamer.py

[Naoki04]

Thank you for publishing this great repository. I noticed a small issue: it seems that the key name of the saved model and the one being loaded in dreamer.py do not match, which leads to a key error.

Specifically, the line:

agent.value_model.load_state_dict(model_dicts['value_model'])

should be:
agent.value_model.load_state_dict(model_dicts['value_model1'])