From 338c76564a1114695fe76637b059bfb9b1a7b7ba Mon Sep 17 00:00:00 2001 From: ceci3 Date: Wed, 14 Oct 2020 22:43:48 +0800 Subject: [PATCH] update syncbn docs, test=document_fix (#27948) --- python/paddle/nn/layer/norm.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/python/paddle/nn/layer/norm.py b/python/paddle/nn/layer/norm.py index ecc89b6c1a..e9a2df3dc6 100644 --- a/python/paddle/nn/layer/norm.py +++ b/python/paddle/nn/layer/norm.py @@ -994,6 +994,11 @@ class SyncBatchNorm(_BatchNormBase): - :math:`\\gamma` : trainable scale parameter vector - :math:`\\beta` : trainable shift parameter vector + Note: + If you want to use container to pack your model and has ``SyncBatchNorm`` in the + evaluation phase, please use ``nn.LayerList`` or ``nn.Sequential`` instead of + ``list`` to pack the model. + Parameters: num_features(int): Indicate the number of channels of the input ``Tensor``. epsilon(float, optional): The small value added to the variance to prevent division by zero. Default: 1e-5.