nami.losses.stochastic_fm

Functions

stochastic_fm_loss(field, x_target, x_source)

Flow matching loss for stochastic interpolants with additive Gaussian noise.

nami.losses.stochastic_fm.stochastic_fm_loss(field, x_target, x_source, t=None, c=None, *, path=None, gamma=None, z=None, reduction='mean')

Flow matching loss for stochastic interpolants with additive Gaussian noise.

Return type:

Tensor

Parameters:

• x_target (Tensor)

• x_source (Tensor)

• t (Tensor | None)

• c (Tensor | None)

• gamma (GammaSchedule | None)

• z (Tensor | None)

• reduction (str)