Closed linjiangya closed 6 months ago
Thanks for pointing this out. I'm still investigating this, but I believe it happened due to the incorrect initialization of the backbone. I have fixed this, and the training losses are decreasing as expected. Could you pull the latest updates and give it another shot?
Thanks for pointing this out. I'm still investigating this, but I believe it happened due to the incorrect initialization of the backbone. I have fixed this, and the training losses are decreasing as expected. Could you pull the latest updates and give it another shot?
I want to train hamer with vitpose_base as the backbone, but how to get the pretrained vitpose_backbone_base.pth?
Originally posted by @linjiangya in https://github.com/geopavlakos/hamer/issues/17#issuecomment-1901318513
Here are examples of using official checkpoints and checkpoints of mine:
epoch=0-step=1000000.ckpt
)hamer.ckpt
)Here's the tensorboard log:![image](https://github.com/geopavlakos/hamer/assets/38423714/93ee5e86-d48e-4814-8bf8-c1cae58d09fd)