Test example doesn't seem to work when scale is set to 4?

[HanYangZhao]

It did work when scale is set to 2, but on 4 i get the following error

    model_cran_v2.load_state_dict(torch.load(checkpoint, 'cpu'))
  File "C:\Users\admin\miniconda3\envs\nvidia-spade\lib\site-packages\torch\nn\modules\module.py", line 1045, in load_state_dict
    self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for Sequential:
        Missing key(s) in state_dict: "1.upsampler.m.1.0.weight", "1.upsampler.m.1.0.bias".