Opened this issue 5 months ago · 1 comments
I have reviewed the code. After running it, it appears that torch.load needs to be applied to 'ckpt_path' in 'state_dict'. Could you provide a pretrained .pth file that I can use to test with test.py?
torch.load
.pth
test.py
You can download the checkpoint using the link now.