yoyo-nb / Thin-Plate-Spline-Motion-Model

[CVPR 2022] Thin-Plate Spline Motion Model for Image Animation.
MIT License
3.38k stars 552 forks source link

Erroring out at - from demo import load_checkpoints #105

Open willtowle opened 2 months ago

willtowle commented 2 months ago

Hey, I was running this in colab yesterday without any issues - but since today myself and a colleague can't run it past a certain point - at this point:

from demo import load_checkpoints
inpainting, kp_detector, dense_motion_network, avd_network = load_checkpoints(config_path = config_path, checkpoint_path = checkpoint_path, device = device)

It errors, the error reads:

FileNotFoundError                         Traceback (most recent call last)
[<ipython-input-13-2c8086808213>](https://localhost:8080/#) in <cell line: 2>()
      1 from demo import load_checkpoints
----> 2 inpainting, kp_detector, dense_motion_network, avd_network = load_checkpoints(config_path = config_path, checkpoint_path = checkpoint_path, device = device)

3 frames
[/usr/local/lib/python3.10/dist-packages/torch/serialization.py](https://localhost:8080/#) in __init__(self, name, mode)
    424 class _open_file(_opener):
    425     def __init__(self, name, mode):
--> 426         super().__init__(open(name, mode))
    427 
    428     def __exit__(self, *args):

FileNotFoundError: [Errno 2] No such file or directory: 'checkpoints/vox.pth.tar'

I've re-traced my steps many times and I definitely haven't missed a step, it looks like no vox.pth.tar is being created at an earlier step.

Any ideas to resolve this?

Thanks