gan_test.cpp File Reference

#include <mlpack/core.hpp>
#include <mlpack/methods/ann/init_rules/gaussian_init.hpp>
#include <mlpack/methods/ann/loss_functions/sigmoid_cross_entropy_error.hpp>
#include <mlpack/methods/ann/gan/gan.hpp>
#include <mlpack/methods/ann/layer/layer.hpp>
#include <mlpack/methods/softmax_regression/softmax_regression.hpp>
#include <ensmallen.hpp>
#include "catch.hpp"
#include "test_catch_tools.hpp"
#include "serialization.hpp"

Include dependency graph for gan_test.cpp:

Functions
	TEST_CASE ("GANTest", "[GANNetworkTest]")
	TEST_CASE ("GANMNISTTest", "[GANNetworkTest]")
	TEST_CASE ("GANMemorySharingTest", "[GANNetworkTest]")

Detailed Description

Author

Kris Singh

Shikhar Jaiswal

Tests the GAN network.

mlpack is free software; you may redistribute it and/or modify it under the terms of the 3-clause BSD license. You should have received a copy of the 3-clause BSD license along with mlpack. If not, see http://www.opensource.org/licenses/BSD-3-Clause for more information.