add test handler

2020-08-24 22:46:36 +08:00
parent 31aafb3470
commit 56b355737f
5 changed files with 65 additions and 9 deletions
--- a/configs/synthesizers/UGATIT.yml
+++ b/configs/synthesizers/UGATIT.yml
@@ -103,10 +103,9 @@ data:
      pin_memory: False
      drop_last: False
    dataset:
-      _type: GenerationUnpairedDataset
-      root_a: "/data/i2i/selfie2anime/testA"
-      root_b: "/data/i2i/selfie2anime/testB"
-      random_pair: False
+      _type: SingleFolderDataset
+      root: "path/to/images/"
+      with_path: True
      pipeline:
        - Load
        - Resize: