fix stylegan (PaddlePaddle#357)

AnnaTrainingG · Jul 7, 2021 · b6f6345 · b6f6345
1 parent 4bb2686
commit b6f6345
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/configs/stylegan_v2_256_ffhq.yaml b/configs/stylegan_v2_256_ffhq.yaml
@@ -45,6 +45,21 @@ dataset:
           - name: Normalize
             mean: [127.5, 127.5, 127.5]
             std: [127.5, 127.5, 127.5]
+  test:
+    name: SingleDataset
+    dataroot: data/ffhq/images256x256/
+    num_workers: 3
+    batch_size: 3
+    preprocess:
+      - name: LoadImageFromFile
+        key: A
+      - name: Transforms
+        input_keys: [A]
+        pipeline:
+          - name: Transpose
+          - name: Normalize
+            mean: [127.5, 127.5, 127.5]
+            std: [127.5, 127.5, 127.5]
 
 lr_scheduler:
   name: MultiStepDecay