Skip to content

The convergence issue during the fine-tuning process #140

@Wangweizhen-S

Description

@Wangweizhen-S

I fine-tuned my dataset using the pi3. I adopted a high-resolution configuration and disabled the confidence head. I noticed that the pose quickly converged, but the points became blurry. However, when using the pre-trained weights of pi3 for prediction, the local point cloud for each frame had a normal shape. Why is that? What are the things to note during the fine-tuning process?

Image(This is the fintuned result.)

Image(This is the pretrained model‘s result.)
The pretraining results have a general shape (perhaps the depth is not precise enough), but the pose is incorrect. Now, after fine-tuning, the results have a more accurate pose, but the depth is lost. And the training loss even still lower, and the validation loss is also much lower than the first few training iterations but remains unchanged after few epochs.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions