spyrit.core.train.Weight_Decay_Loss

class spyrit.core.train.Weight_Decay_Loss(loss)[source]

Bases: Module

Methods

forward(x, y, net)

Define the computation performed at every call.