compute_loss

Contents

compute_loss#

ptyrad.reconstruction.compute_loss(batch, model, model_instance, loss_fn, acc=None)[source]#

Compute the model output and loss, with optional support for accelerate’s autocast.