MolecularDiffusion.modules.models.tabasco.utils.metric_utils¶

Attributes¶

log

Functions¶

split_losses_by_time(→ List[float])

Split the losses by time.

Module Contents¶

MolecularDiffusion.modules.models.tabasco.utils.metric_utils.split_losses_by_time(flow_time: torch.Tensor, losses: torch.Tensor, num_bins: int) → List[float]¶

Split the losses by time.

Parameters:

time – time tensor
losses – losses tensor
num_bins – number of bins

Returns:

list of mean losses for each bin

Return type:

List[float]

MolecularDiffusion.modules.models.tabasco.utils.metric_utils.log¶