提交 bf08e5d9 编写于 作者: D dangqingqing

modify code comments.

上级 73192bb1
......@@ -20,20 +20,18 @@ limitations under the License. */
/**
* @brief batch norm inferece.
*
* @param[in] input input data.
* @param[out] output output data.
* @param[in] scale batch normalization scale parameter (in original
* paper scale is referred to as gamma).
* @param[in] bias batch normalization bias parameter (in original
* paper scale is referred to as beta).
* @param[in] input input data.
* @param[out] output output data.
* @param[in] scale batch normalization scale parameter (in original
* paper scale is referred to as gamma).
* @param[in] bias batch normalization bias parameter (in original
* paper scale is referred to as beta).
* @param[in] estimatedMean
* @param[in] estimatedVar It is suggested that resultRunningMean,
* resultRunningVariance from the
* cudnnBatchNormalizationForwardTraining call
* accumulated during the training phase are passed
* as inputs here.
* @param[in] epsilon Epsilon value used in the batch
* normalization formula.
* @param[in] estimatedVar The moving mean and variance
* accumulated during the training phase are passed
* as inputs here.
* @param[in] epsilon Epsilon value used in the batch
* normalization formula.
*/
extern void hl_batch_norm_cuda_inference(const real* input,
real* output,
......
......@@ -81,8 +81,8 @@ void CudnnBatchNormLayer::forward(PassType passType) {
} else {
// used movingMean and movingVar in testing
if (batchSize > 1024) {
// when batchSize is larger than 1024, there is a bug
// in cudnn library.
// there is a bug in cudnn library when the batch size
// is larger than 1024.
hl_batch_norm_cuda_inference(input,
output,
gamma,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册