def __init__(self, input_shape, log_dir, filters=32, kernel_size=2, latent_channels=1, beta=1.0):
# initialise FreyVAE specific variables
self.filters = filters
self.kernel_size = kernel_size
self.latent_channels = latent_channels
# call parent constructor
VAE.__init__(self, input_shape, log_dir, beta=beta)
def __init__(self, input_shape, log_dir, filters=32, kernel_size=2, pre_latent_size=64, latent_size=2):
# initialise FreyVAE specific variables
self.filters = filters
self.kernel_size = kernel_size
self.pre_latent_size = pre_latent_size
self.latent_size = latent_size
# call parent constructor
VAE.__init__(self, input_shape, log_dir)
def __init__(self,
input_shape,
log_dir,
filters=32,
latent_filters=64,
kernel_size=6,
pool_size=2):
# initialise ConvolutionalLatentVAE specific variables
self.filters = filters
self.latent_filters = latent_filters
self.kernel_size = kernel_size
self.pool_size = pool_size
# call parent constructor
VAE.__init__(self, input_shape, log_dir)
def __init__(self,
input_shape,
log_dir,
filters=32,
pool_size=(2, 2),
kernel_size=6,
pre_latent_size=128,
latent_size=32,
beta=1.0):
self.input_shape = input_shape
self.filters = filters
self.pool_size = pool_size
self.kernel_size = kernel_size
self.pre_latent_size = pre_latent_size
self.latent_size = latent_size
VAE.__init__(self, input_shape, log_dir, beta=beta)
