Skip to content

[BUG] Training stage is silently ignored in CouplingFlow.compute_metrics #533

@han-ol

Description

@han-ol

Currently, compute_metrics in CouplingFlow does NOT overwrite the default training=False in self.call for forward passes, even during training.
One side effect is that dropout layers are silently ignored.

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions