Adriano D.
Adriano D.
Hello refer to issue https://github.com/JunkyByte/easy_ViTPose/issues/15
Provide more informations. Are you using matching model sizes for checkpoint and model declaration?
The models are split into the different heads using `https://github.com/JunkyByte/easy_ViTPose/blob/main/model_split.py` The models provided in hugging face are taken from the original section with this process applied. Try using the model_split...
Hello! Which code did you use for training? Are you able to run inference successfully with such code? Could you share the inference code and skeleton you are using?
Cool! I'm currently in holiday but will check the code change when I come back. Have a nice day!
Hello! Sorry for this late reply, I'm on vacation... Check the discussion on https://github.com/JunkyByte/easy_ViTPose/issues/8#issuecomment-1671773308 You have to use vitpose+ ckpts, model split should work correctly. After you get the ckpt...
Same bug with pytorch-lightning 2.5.0.post0 torch 2.5.1