dgs.models.loss

Functions to load and manage torch and custom loss functions.

Module Functions

get_loss_function(instance)

Given the name or an instance of a loss function, return the respective instance.

register_loss_function(name, new_loss)

Register a new loss function to be used with custom configs.

Modules

loss

Custom loss functions.