prediction

[haifengwu205]

When I run the test, the mode loading shape does not match

Traceback (most recent call last):
File "trainer.py", line 228, in
main()
File "trainer.py", line 149, in main
model = DepthLitModel.load_from_checkpoint(CKPT_PATH, config=config)
File "/data/home/wuhaifeng/anaconda3/envs/p3depth/lib/python3.7/site-packages/pytorch_lightning/core/saving.py", line 157, in load_from_checkpoint
model = cls._load_model_state(checkpoint, strict=strict, **kwargs)
File "/data/home/wuhaifeng/anaconda3/envs/p3depth/lib/python3.7/site-packages/pytorch_lightning/core/saving.py", line 205, in _load_model_state
model.load_state_dict(checkpoint['state_dict'], strict=strict)
File "/data/home/wuhaifeng/anaconda3/envs/p3depth/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1672, in load_state_dict
self.class.name, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for DepthLitModel:
size mismatch for net.coords: copying a param with shape torch.Size([8, 480, 640, 2]) from checkpoint, the shape in current model is torch.Size([16, 480, 640, 2]).