From b1806b8432a43b1d892cee588701630bc94db25b Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Fri, 22 Jan 2021 12:13:40 +0800
Subject: [PATCH] docs(mgb/opr): improve batch_norm err msg in training mode

GitOrigin-RevId: 4c43bc8b0b08605238b8eaa0a1ce67d2dae5e348
---
 src/opr/impl/dnn/batch_norm.cpp | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/opr/impl/dnn/batch_norm.cpp b/src/opr/impl/dnn/batch_norm.cpp
index d58c17f6..73fd6956 100644
--- a/src/opr/impl/dnn/batch_norm.cpp
+++ b/src/opr/impl/dnn/batch_norm.cpp
@@ -50,8 +50,9 @@ BatchNormForward::BatchNormForward(VarNode *x,
             mgb_throw_if(!(dest_opr->same_type<SharedDeviceTensor>() ||
                     dest_opr->same_type<VolatileSharedDeviceTensor>()),
                     GraphError,
-                    "mean and variance in BatchNorm must be SharedDeviceTensor "
-                    "or VolatileSharedDeviceTensor; got %s{%s} actually",
+                    "mean and variance in training mode BatchNorm must be",
+                    "SharedDeviceTensor or VolatileSharedDeviceTensor;",
+                    "got %s{%s} actually",
                     dest_opr->cname(), dest_opr->dyn_typeinfo()->name);
         };
         check_dest(mean);
-- 
GitLab