Convert between ".pt" and ".ckpt"

huawei-noah / Pretrained-IPT

Apache License 2.0

424 stars 63 forks source link

Convert between ".pt" and ".ckpt" #24

Closed abood002 closed 1 year ago

abood002 commented 2 years ago

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.

HantingChen commented 2 years ago

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