loss_function

FastProp.loss_function: Optional[Literal[CrossEntropyLoss, SquareLoss]]