diff --git a/src/mlpack/tests/custom_layer.hpp b/src/mlpack/tests/custom_layer.hpp index a8ad5cf4a2..8d791b1ebd 100644 --- a/src/mlpack/tests/custom_layer.hpp +++ b/src/mlpack/tests/custom_layer.hpp @@ -2,7 +2,7 @@ * @file tests/custom_layer.hpp * @author Projyal Dev * - * A simple custom layer mimicing SigmoidLayer for testing if custom + * A simple custom layer mimicking SigmoidLayer for testing if custom * layers work. * * mlpack is free software; you may redistribute it and/or modify it under the @@ -20,16 +20,17 @@ namespace mlpack { namespace ann { - /** - * Standard Sigmoid layer. - */ - template < - class ActivationFunction = LogisticFunction, - typename InputDataType = arma::mat, - typename OutputDataType = arma::mat - > - using CustomLayer = BaseLayer< - ActivationFunction, InputDataType, OutputDataType>; + +/** + * Standard Sigmoid layer. + */ +template < + class ActivationFunction = LogisticFunction, + typename InputDataType = arma::mat, + typename OutputDataType = arma::mat +> +using CustomLayer = BaseLayer< + ActivationFunction, InputDataType, OutputDataType>; } // namespace ann } // namespace mlpack