pals-ttic / sjc

Score Jacobian Chaining: Lifting Pretrained 2D Diffusion Models for 3D Generation (CVPR 2023)
https://pals.ttic.edu/p/score-jacobian-chaining
Other
500 stars 15 forks source link

Where is the jacobian implementation for the backpropogation? #22

Closed canxkoz closed 1 year ago

canxkoz commented 1 year ago

I could not find the jacobian implementation for the backpropagation. I need to use the jacobian loss function which is shared in the paper, however I cannot find it. Could anyone please help me find where the loss function is implemented? image_6487327 Above is a screenshot taken from the paper. I am searching for the implemetation of the chain rule in code form. Could you please help me find where that specific function is? Thank you for your time,