-
Notifications
You must be signed in to change notification settings - Fork 0
Open
Description
In contrast to Processor.validation_step – which should mostly be the same for all processors (compares rolled out trajectory to true trajectory) – Processor.training_step seems to depend more on the specific subclass of Processor. E.g., the flow matching loss can be different from the diffusion loss.
Related to this: I think we can get rid of
| loss_func: nn.Module | None = None, |
as, in most cases, the loss function will be specific to the subclass.
sgreenbury
Metadata
Metadata
Assignees
Labels
No labels
Type
Projects
Status
Todo