Browse Source

docs(mgb/opr): improve batch_norm err msg in training mode

GitOrigin-RevId: 4c43bc8b0b
tags/v1.3.0
Megvii Engine Team 4 years ago
parent
commit
b1806b8432
1 changed files with 3 additions and 2 deletions
  1. +3
    -2
      src/opr/impl/dnn/batch_norm.cpp

+ 3
- 2
src/opr/impl/dnn/batch_norm.cpp View File

@@ -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);


Loading…
Cancel
Save