in privacy_lint/attacks/loss.py [0:0]
def __init__( self, compute_loss: Callable[ [nn.Module, DataLoader], torch.Tensor ] = compute_loss_cross_entropy,