{ "dataset_kwargs": { "class_name": "training.dataset.ImageFolderDataset", "path": "datasets/cifar10-32x32.zip", "use_labels": false, "xflip": false, "cache": true, "resolution": 32, "max_size": 50000 }, "data_loader_kwargs": { "pin_memory": true, "num_workers": 1, "prefetch_factor": 2 }, "network_kwargs": { "model_type": "SongUNet", "embedding_type": "positional", "encoder_type": "standard", "decoder_type": "standard", "channel_mult_noise": 1, "resample_filter": [ 1, 1 ], "model_channels": 128, "channel_mult": [ 2, 2, 2 ], "class_name": "training.networks.BetaDiffPrecond", "augment_dim": 9, "dropout": 0.13, "use_fp16": false }, "loss_kwargs": { "class_name": "training.loss.BetaDiffLoss", "eta": 10000.0, "sigmoid_start": 10.0, "sigmoid_end": -13.0, "sigmoid_power": 1.0, "Scale": 0.39, "Shift": 0.6, "lossType": "KLUB" }, "optimizer_kwargs": { "class_name": "torch.optim.Adam", "lr": 0.0002, "betas": [ 0.9, 0.999 ], "eps": 1e-08 }, "augment_kwargs": { "class_name": "training.augment.AugmentPipe", "p": 0.12, "xflip": 100000000.0, "yflip": 1, "scale": 1, "rotate_frac": 1, "aniso": 1, "translate_frac": 1 }, "total_kimg": 200000, "ema_halflife_kimg": 500, "batch_size": 288, "batch_gpu": null, "loss_scaling": 1.0, "cudnn_benchmark": true, "kimg_per_tick": 50, "snapshot_ticks": 50, "state_dump_ticks": 500, "seed": 1139385602, "resume_pkl": "betadiff-train-runs/00000-cifar10-32x32-uncond-ddpmpp-betadiff-gpus3-batch288-fp32/network-snapshot-100224.pkl", "resume_kimg": 100224, "resume_state_dump": "betadiff-train-runs/00000-cifar10-32x32-uncond-ddpmpp-betadiff-gpus3-batch288-fp32/training-state-100224.pt", "run_dir": "betadiff-train-runs/00001-cifar10-32x32-uncond-ddpmpp-betadiff-gpus3-batch288-fp32" }