diff --git a/src/mlpack/methods/reinforcement_learning/q_networks/categorical_dqn.hpp b/src/mlpack/methods/reinforcement_learning/q_networks/categorical_dqn.hpp index d6f5374096..917636d87b 100644 --- a/src/mlpack/methods/reinforcement_learning/q_networks/categorical_dqn.hpp +++ b/src/mlpack/methods/reinforcement_learning/q_networks/categorical_dqn.hpp @@ -41,9 +41,9 @@ namespace rl { * @tparam NetworkType The type of network used for simple dqn. */ template< - typename OutputLayerType = mlpack::ann::EmptyLoss<>, - typename InitType = mlpack::ann::GaussianInitialization, - typename NetworkType = mlpack::ann::FFN + typename OutputLayerType = ann::EmptyLoss<>, + typename InitType = ann::GaussianInitialization, + typename NetworkType = ann::FFN > class CategoricalDQN {