Checkpoints Mismatch

[csqiangwen]

Hi, thanks for sharing this excellent work.

I met a checkpoints mismatch problem.

When I want to run the 'test_fence.py' file, I load the checkpoints under the folder shown in the above pic.
And I also modify the 'ckpt_path' arguments in the 'test_fence.py' file, as shown in the following:

But I still meet this problem :

I have checked the pwcnet checkpoints could be loaded successfully.
Could you help me solve this problem?