Pretrained model-Unexpected key
Mareeta26 opened this issue · 1 comments
Mareeta26 commented
Hi,
I tried testing the pretrained model of predrnn-v2 for moving-mnist dataset. But I get the following error. Can you please check the reason?
RuntimeError: Error(s) in loading state_dict for RNN:
Unexpected key(s) in state_dict: "adapter.weight".
Thanks!!
Mareeta26 commented
Found that I needed to edit the model name and the hyper parameters