diff --git a/nnvm/include/nnvm/top/nn.h b/nnvm/include/nnvm/top/nn.h index 0e0b03a72ebb..69d49cd99696 100644 --- a/nnvm/include/nnvm/top/nn.h +++ b/nnvm/include/nnvm/top/nn.h @@ -349,7 +349,7 @@ struct MultiBoxTransformLocParam : public dmlc::Parameter{0.1, 0.1, 0.2, 0.2}) + DMLC_DECLARE_FIELD(variances).set_default(Tuple({0.1f, 0.1f, 0.2f, 0.2f})) .describe("Variances to be decoded from box regression output."); } };