liuyuan-pal/Gen6D

Selector and refiner training issue

EvdoTheo opened this issue · 1 comments

Hello @liuyuan-pal, I'm trying to train the selector on my custom data and the loss score is not decreasing, instead, it is stable at 0.4. What do you suggest in order to unstuck it?

Moreover, I also try to train the refiner, but the new refiner is giving me worse results than the pretrained one. For example, when the refiner starts working after a few frames it completely loses its target and the detection algorithm must be deployed to initialize pose. I attached the inter folder to give it a glance. Thank you!

Hi, I'm not sure about the problem here. We may need to visualize the intermediate results and check the model is able to overfit a small training set for sanity check purposes.