{ "clip": 1.0, "cls_epsilon": 0.05, "cls_ema_window": 32, "augmentations_before_corruption": true, "overwrite_cls_labels_path": null, "crop_size": null, "dataset_kwargs": { "path": "./data/cifar10/train", "use_labels": false, "cache": true, "corruptions_dict": null, "corruption_probability": 0.5, "only_positive": false, "dataset_keep_percentage": 1.0, "resolution": 32, "max_size": 50000, "noise_config": "blur0-6" }, "data_loader_kwargs": { "pin_memory": true, "num_workers": 2, "prefetch_factor": 8 }, "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.ambient_networks.EDMPrecondCLS", "num_classes": 1, "augment_dim": 9, "dropout": 0.13, "use_fp16": false }, "loss_kwargs": { "class_name": "training.loss.AmbientEDMCLSLoss" }, "sampler_kwargs": {}, "optimizer_kwargs": { "class_name": "torch.optim.Adam", "lr": 0.0001, "betas": [ 0.9, 0.999 ], "eps": 1e-08, "weight_decay": 0.0 }, "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": 512, "batch_gpu": null, "loss_scaling": 1.0, "cudnn_benchmark": true, "kimg_per_tick": 50, "snapshot_ticks": 20, "state_dump_ticks": 40, "seed": 743050150, "run_dir": "./outputs/ambient-syn-runs/cifar/noise_classifier/blur0-6_prob0-5/00000-train-uncond-ddpmpp-edmcls-gpus8-batch512-fp32-TWeeb" }