added deepsad base code
This commit is contained in:
10
Deep-SAD-PyTorch/src/networks/__init__.py
Normal file
10
Deep-SAD-PyTorch/src/networks/__init__.py
Normal file
@@ -0,0 +1,10 @@
|
||||
from .main import build_network, build_autoencoder
|
||||
from .mnist_LeNet import MNIST_LeNet, MNIST_LeNet_Decoder, MNIST_LeNet_Autoencoder
|
||||
from .fmnist_LeNet import FashionMNIST_LeNet, FashionMNIST_LeNet_Decoder, FashionMNIST_LeNet_Autoencoder
|
||||
from .cifar10_LeNet import CIFAR10_LeNet, CIFAR10_LeNet_Decoder, CIFAR10_LeNet_Autoencoder
|
||||
from .mlp import MLP, MLP_Decoder, MLP_Autoencoder
|
||||
from .layers.stochastic import GaussianSample
|
||||
from .layers.standard import Standardize
|
||||
from .inference.distributions import log_standard_gaussian, log_gaussian, log_standard_categorical
|
||||
from .vae import VariationalAutoencoder, Encoder, Decoder
|
||||
from .dgm import DeepGenerativeModel, StackedDeepGenerativeModel
|
||||
Reference in New Issue
Block a user