configs/cifar10_k_ve.py [5:31]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
def get_config():
    config = get_default_configs()
    # training
    training = config.training
    training.sde = "kvesde"
    training.loss = "dsm"
    training.batch_size = 512
    training.n_iters = 400001
    training.n_jitted_steps = 2
    training.snapshot_freq = 10000
    training.snapshot_freq_for_preemption = 5000
    training.log_freq = 50
    training.eval_freq = 100

    # sampling
    sampling = config.sampling
    sampling.method = "heun"
    sampling.denoise = True

    # evaluation
    evaluate = config.eval
    evaluate.begin_ckpt = 1
    evaluate.end_ckpt = 40

    # model
    model = config.model
    model.name = "ncsnpp"
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



configs/cifar10_k_vp.py [5:32]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
def get_config():
    config = get_default_configs()
    # training
    training = config.training
    training.sde = "kvesde"
    training.loss = "dsm"
    training.batch_size = 512
    training.n_iters = 400001
    training.n_jitted_steps = 2
    training.snapshot_freq = 10000
    training.snapshot_freq_for_preemption = 5000
    training.log_freq = 50
    training.eval_freq = 100

    # sampling
    sampling = config.sampling
    sampling.method = "heun"
    sampling.denoise = True

    # evaluation
    evaluate = config.eval
    evaluate.begin_ckpt = 1
    evaluate.end_ckpt = 40

    # model
    model = config.model

    model.name = "ncsnpp"
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -