nnspike.models.loss

Classes

MultiTaskLoss([mode_weight, control_weight, ...])

class nnspike.models.loss.MultiTaskLoss(mode_weight=1.0, control_weight=30.0, control_scale=10.0)[source]

__init__(mode_weight=1.0, control_weight=30.0, control_scale=10.0)[source]

forward(outputs, targets)[source]