From ed3b9ae7b0f3658fad0d10cdf4364020fa03ac3c Mon Sep 17 00:00:00 2001 From: Eldar Kurtic Date: Thu, 25 Mar 2021 18:51:31 +0100 Subject: [PATCH] Typo? --- src/sparseml/pytorch/utils/loss.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/sparseml/pytorch/utils/loss.py b/src/sparseml/pytorch/utils/loss.py index 65d9aaf288d..ac8aae1c2bb 100644 --- a/src/sparseml/pytorch/utils/loss.py +++ b/src/sparseml/pytorch/utils/loss.py @@ -220,7 +220,7 @@ def __init__( class InceptionCrossEntropyLossWrapper(LossWrapper): """ - Loss wrapper for training an inception model that as an aux output + Loss wrapper for training an inception model that has an aux output with cross entropy. Defines the loss in the following way: