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)