How to load checkpoint(.ckpt) in test code?

JiayuYANG / CVP-MVSNet

Cost Volume Pyramid Based Depth Inference for Multi-View Stereo (CVPR 2020 Oral)

239 stars 33 forks source link

How to load checkpoint(.ckpt) in test code? #29

Open min-jung-shin opened 3 years ago

min-jung-shin commented 3 years ago

hello. While I run this code, I'can't load test checkpoint file you uploaded. It doesn't use tensorflow but, it is saved .ckpt file extension. How to do it?

JiayuYANG commented 3 years ago

Hi @min-jung-shin

Even though the file extension is .ckpt, it is actually a PyTorch checkpoint and can not be directly used in Tensorflow. I used this line to load the checkpoint in PyTorch.

Jiayu.