mypackage.MyModule.loss

MyModule.loss(tensors, inference_outputs, generative_outputs, kl_weight=1.0)

Compute the loss for a minibatch of data.

This function uses the outputs of the inference and generative functions to compute a loss. This many optionally include other penalty terms, which should be computed here.

This function should return an object of type LossRecorder.