search

jasonyzhang / RayDiffusion

Code for "Cameras as Rays"
MIT License
483 stars 23 forks source link

predict_x0 parameter #17

Closed liziwei-0 closed 3 months ago

liziwei-0 commented 3 months ago

Hello, thank you for your great work on this project!  I have a question regarding the predict_x0 parameter. Specifically, I am trying to understand its purpose and how it influences the training process. From what I can gather, it seems to determine the target during training, but I would like to get more clarity on this.

liziwei-0 commented 3 months ago

I am currently working on a similar task and would like to better understand the training process and results. Could you please provide the training logs for reference?

jasonyzhang commented 3 months ago

For diffusion, predict_x0 trains a model that does x0 prediction instead of epsilon prediction. The regression model should always do x0 prediction since it's a regression model.