DCGAN on CelebA Dataset using Libtorch (PyTorch C++ Frontend API)
|
#include <iostream>
#include <torch/torch.h>
#include <opencv2/opencv.hpp>
#include <torch/script.h>
Go to the source code of this file.
Classes | |
struct | ConvTranspose2dWrapperImpl |
class | Generator |
class | Discriminator |
Functions | |
TORCH_MODULE (ConvTranspose2dWrapper) | |
TORCH_MODULE | ( | ConvTranspose2dWrapper | ) |