From 5265c90884f91a6b4a2b297723925ac0ff324aab Mon Sep 17 00:00:00 2001 From: gengdongjie Date: Tue, 28 Apr 2020 22:32:49 +0800 Subject: [PATCH] add reduce mean in SoftmaxCrossEntropyWithLogits for resent50 example --- example/resnet50_cifar10/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/example/resnet50_cifar10/train.py b/example/resnet50_cifar10/train.py index 0a0299b2bdd..4bae682ba04 100755 --- a/example/resnet50_cifar10/train.py +++ b/example/resnet50_cifar10/train.py @@ -68,7 +68,7 @@ if __name__ == '__main__': epoch_size = config.epoch_size net = resnet50(class_num=config.class_num) - loss = SoftmaxCrossEntropyWithLogits(sparse=True) + loss = SoftmaxCrossEntropyWithLogits(sparse=True, reduction="mean") if args_opt.do_train: