restore UGATIT.yml

2020-08-25 09:33:42 +08:00
parent 56b355737f
commit 09db0a413f
1 changed files with 4 additions and 3 deletions
--- a/configs/synthesizers/UGATIT.yml
+++ b/configs/synthesizers/UGATIT.yml
@@ -103,9 +103,10 @@ data:
      pin_memory: False
      drop_last: False
    dataset:
-      _type: SingleFolderDataset
-      root: "path/to/images/"
-      with_path: True
+      _type: GenerationUnpairedDataset
+      root_a: "/data/i2i/selfie2anime/testA"
+      root_b: "/data/i2i/selfie2anime/testB"
+      random_pair: False
      pipeline:
        - Load
        - Resize: