Resume from a pre-trained model

[primecai]

Hello,

Many thanks for releasing the code. I'm wondering if the code actually support resuming from a checkpoint file? It looks like the function parse_train_file in kp2d/utils/config.py rejects a .ckpt file.

Best regards.

[RaresAmbrus]

Hi, thanks for the question. You're right, that function looks incomplete -- I may have deleted parts of it while cleaning up the code. We have some code that loads a .ckpt file in the eval script (here) which could be re-purposed for that. I'm a bit busy these days and can't look into this right away, but if you manage to get this working and can push a fix I would be very happy :-)