search

SKholkin / LightSB-Matching

Light and Optimal Schrödinger Bridge Matching (ICML 2024) official PyTorch implementation=
MIT License
28 stars 4 forks source link

Loss plan #1

Open kaiyamchu opened 1 month ago

kaiyamchu commented 1 month ago

Hi, Here is Jiaxin.

May I know what the loss_plan w.r.t. is in the code?

SKholkin commented 1 month ago

Hi, loss_plan is a loss value holder corresponding to another method of fitting LightSB-M model parameters, also known as LightSB loss. It is equivalent to the Optimal Projection Loss up to a constant (see https://arxiv.org/abs/2402.03207 Th. 3.3).