huawei-noah/Pretrained-IPT

Convert between ".pt" and ".ckpt"

abood002 opened this issue · 1 comments

Hello,
Thanks for sharing your great work.
I wanted to ask if there is a way to convert the pre-trained model from ".pt" format to ".ckpt" and from ".ckpt" format to ".pt".
Thanks in advance.

You can directly renname the checkpoint file as .pt instead of .ckpt.