Skip to content

The convergence issue during the fine-tuning process #140

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.

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

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