From 09db0a413f8a1c10564571583c07e0b1da3b4dd4 Mon Sep 17 00:00:00 2001
From: budui <wangbudui@foxmail.com>
Date: Tue, 25 Aug 2020 09:33:42 +0800
Subject: [PATCH] restore UGATIT.yml

---
 configs/synthesizers/UGATIT.yml | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/configs/synthesizers/UGATIT.yml b/configs/synthesizers/UGATIT.yml
index 07160fe..05d5311 100644
--- 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: