Composable

Module: generative_models.core.losses.composable

Source: generative_models/core/losses/composable.py

Overview

Composable loss function framework using Flax NNX.

Classes

CompositeLoss

class CompositeLoss

Loss

class Loss

ScheduledLoss

class ScheduledLoss

WeightedLoss

class WeightedLoss

Functions

call

def __call__()

call

def __call__()

call

def __call__()

call

def __call__()

init

def __init__()

init

def __init__()

init

def __init__()

init

def __init__()

create_loss_suite

def create_loss_suite()

create_weighted_loss

def create_weighted_loss()

weighted_loss_fn

def weighted_loss_fn()

Module Statistics

• Classes: 4
• Functions: 11
• Imports: 4