def get_config(): config = get_default_configs() # training training = config.training training.sde = 'vesde' training.continuous = True # sampling sampling = config.sampling sampling.method = 'pc' sampling.predictor = 'reverse_diffusion' sampling.corrector = 'langevin' # data data = config.data data.dataset = 'CUSTOM' data.image_size = 128 training.batch_size = 16 data.tfrecords_path = "/content/drive/MyDrive/Training/tf_dataset/tf_dataset-r07.tfrecords" # model model = config.model model.name = 'ncsnpp' model.sigma_max = 217 model.scale_by_sigma = True model.ema_rate = 0.999 model.normalization = 'GroupNorm' model.nonlinearity = 'swish' model.nf = 128 model.ch_mult = (1, 1, 2, 2, 2, 2, 2) model.num_res_blocks = 2 model.attn_resolutions = (16, ) model.resamp_with_conv = True model.conditional = True model.fir = True model.fir_kernel = [1, 3, 3, 1] model.skip_rescale = True model.resblock_type = 'biggan' model.progressive = 'output_skip' model.progressive_input = 'input_skip' model.progressive_combine = 'sum' model.attention_type = 'ddpm' model.init_scale = 0. model.fourier_scale = 16 model.conv_size = 3 return config
def get_config(): config = get_default_configs() # training training = config.training training.sde = 'vpsde' training.continuous = False training.reduce_mean = True # sampling sampling = config.sampling sampling.method = 'pc' sampling.predictor = 'ancestral_sampling' sampling.corrector = 'none' # data data = config.data data.dataset = 'CelebAHQ' data.centered = True data.tfrecords_path = '/atlas/u/yangsong/celeba_hq/-r10.tfrecords' data.image_size = 256 # model model = config.model model.name = 'ddpm' model.scale_by_sigma = False model.num_scales = 1000 model.ema_rate = 0.9999 model.normalization = 'GroupNorm' model.nonlinearity = 'swish' model.nf = 128 model.ch_mult = (1, 1, 2, 2, 4, 4) model.num_res_blocks = 2 model.attn_resolutions = (16, ) model.resamp_with_conv = True model.conditional = True # optim optim = config.optim optim.lr = 2e-5 return config
def get_config(): config = get_default_configs() # training training = config.training training.batch_size = 128 training.sde = 'vesde' training.continuouse = False # sampling sampling = config.sampling sampling.method = 'pc' sampling.predictor = 'none' sampling.corrector = 'ald' sampling.n_steps_each = 3 sampling.snr = 0.095 # data data = config.data data.category = 'bedroom' data.image_size = 128 # model model = config.model model.name = 'ncsnv2_128' model.scale_by_sigma = True model.sigma_max = 190 model.num_scales = 1086 model.ema_rate = 0.9999 model.sigma_min = 0.01 model.normalization = 'InstanceNorm++' model.nonlinearity = 'elu' model.nf = 128 model.interpolation = 'bilinear' # optim optim = config.optim optim.weight_decay = 0 optim.optimizer = 'Adam' optim.lr = 1e-4 optim.beta1 = 0.9 optim.amsgrad = False optim.eps = 1e-8 optim.warmup = 0 optim.grad_clip = -1 return config
def get_config(): config = get_default_configs() # training training = config.training training.sde = 'vpsde' training.continuous = False training.reduce_mean = True # sampling sampling = config.sampling sampling.method = 'pc' sampling.predictor = 'ancestral_sampling' sampling.corrector = 'none' # data data = config.data data.category = 'church_outdoor' data.centered = True # model model = config.model model.name = 'ddpm' model.scale_by_sigma = False model.num_scales = 1000 model.ema_rate = 0.9999 model.normalization = 'GroupNorm' model.nonlinearity = 'swish' model.nf = 128 model.ch_mult = (1, 1, 2, 2, 4, 4) model.num_res_blocks = 2 model.attn_resolutions = (16, ) model.resamp_with_conv = True model.conditional = True # optim optim = config.optim optim.lr = 2e-5 return config