how to load the pth file and inference

[smileto1]

Hello, your work is excellent, but it seems that I haven't found the steps of your model inference process, so I can't load the saved parameters and infer a series images. And the args command parsing doesn't have the checkpoint_path option, but I saw it in your GitHub.

[smileto1]

i have solved this problem