modif to run only on one noise scale

pablo2909 · Sep 11, 2023 · 7c7d348 · 7c7d348
1 parent 230cf83
commit 7c7d348
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/configs/anneal.yml b/configs/anneal.yml
@@ -29,9 +29,9 @@ data:
 # random_flip: true
 
 model:
- sigma_begin: 1
- sigma_end: 0.01
- num_classes: 10
+ sigma_begin: 0.1
+ sigma_end: 0.1
+ num_classes: 1
  batch_norm: false
  ## configurations for CelebA, CIFAR10
 # ngf: 128