diff --git a/src/mlpack/methods/ann/gan/gan.hpp b/src/mlpack/methods/ann/gan/gan.hpp index a88b21e0d0..0fa9fde744 100644 --- a/src/mlpack/methods/ann/gan/gan.hpp +++ b/src/mlpack/methods/ann/gan/gan.hpp @@ -105,8 +105,8 @@ class GAN * @return The final objective of the trained model (NaN or Inf on error). */ template - double Train(OptimizerType& Optimizer, - CallbackTypes&&... callbacks); + double Train(OptimizerType& Optimizer, + CallbackTypes&&... callbacks); /** * Evaluate function for the Standard GAN and DCGAN. diff --git a/src/mlpack/methods/ann/gan/gan_impl.hpp b/src/mlpack/methods/ann/gan/gan_impl.hpp index 20e9407de6..ed822b6904 100644 --- a/src/mlpack/methods/ann/gan/gan_impl.hpp +++ b/src/mlpack/methods/ann/gan/gan_impl.hpp @@ -191,8 +191,7 @@ template< typename PolicyType > template -double GAN::Train( - OptimizerType& Optimizer, CallbackTypes&&... callbacks) +double GAN::Train(OptimizerType& Optimizer, CallbackTypes&&... callbacks) { if (!reset) Reset();