diff --git a/configs/cond_dcgan_mnist.yaml b/configs/cond_dcgan_mnist.yaml
index 0c3aba181863b481ba44720d376973f761e855e3..e708396587b4a6271a7fe467f24613526468d880 100644
--- a/configs/cond_dcgan_mnist.yaml
+++ b/configs/cond_dcgan_mnist.yaml
@@ -18,46 +18,57 @@ model:
     norm_type: batch
     n_class: 10
     use_sigmoid: True
-  gan_mode: vanilla
+  gan_criterion:
+    name: GANLoss
+    gan_mode: vanilla
 
 dataset:
   train:
     name: CommonVisionDataset
-    class_name: MNIST
-    dataroot: None
+    dataset_name: MNIST
     num_workers: 4
     batch_size: 64
-    mode: train
-    return_cls: True
+    return_label: True
     transforms:
       - name: Normalize
         mean: [127.5]
         std: [127.5]
         keys: [image]
+    params:
+      mode: train
   test:
     name: CommonVisionDataset
-    class_name: MNIST
-    dataroot: None
+    dataset_name: MNIST
     num_workers: 0
     batch_size: 64
-    mode: test
+    return_label: True
+    params:
+      mode: train
     transforms:
       - name: Normalize
         mean: [127.5]
         std: [127.5]
         keys: [image]
-    return_cls: True
-
-
-optimizer:
-  name: Adam
-  beta1: 0.5
 
 lr_scheduler:
-  name: linear
+  name: LinearDecay
   learning_rate: 0.0002
   start_epoch: 100
   decay_epochs: 100
+  # will get from real dataset
+  iters_per_epoch: 1
+
+optimizer:
+  optimizer_G:
+    name: Adam
+    net_names:
+      - netG
+    beta1: 0.5
+  optimizer_D:
+    name: Adam
+    net_names:
+      - netD
+    beta1: 0.5
 
 log_config:
   interval: 100
diff --git a/configs/pix2pix_facades.yaml b/configs/pix2pix_facades.yaml
index 0123bf9afc4e17108055c9c6c7f23a9e5acc6ddf..b73005dfe71eee476de08e37ca176e6f9ffa0c2c 100644
--- a/configs/pix2pix_facades.yaml
+++ b/configs/pix2pix_facades.yaml
@@ -64,6 +64,9 @@ dataset:
     preprocess:
       - name: LoadImageFromFile
         key: pair
+      - name: SplitPairedImage
+        key: pair
+        paired_keys: [A, B]
       - name: Transforms
         input_keys: [A, B]
         pipeline:
diff --git a/ppgan/apps/animegan_predictor.py b/ppgan/apps/animegan_predictor.py
index 8c5655d674c10a2f04c39d8d40246c85c9ca5404..b3c8b0b57975efd3250303c468ce9f60389fdd08 100644
--- a/ppgan/apps/animegan_predictor.py
+++ b/ppgan/apps/animegan_predictor.py
@@ -18,7 +18,7 @@ import cv2
 
 import paddle
 from .base_predictor import BasePredictor
-from ppgan.datasets.transforms import ResizeToScale
+from ppgan.datasets.preprocess.transforms import ResizeToScale
 import paddle.vision.transforms as T
 from ppgan.models.generators import AnimeGenerator
 from ppgan.utils.download import get_path_from_url
diff --git a/ppgan/datasets/preprocess/transforms.py b/ppgan/datasets/preprocess/transforms.py
index 73481e84ae7b220b28c244ba738ff21c96a40f17..ab378c2b9c9e2ce7aa05b9ed5565e962cf42d11d 100644
--- a/ppgan/datasets/preprocess/transforms.py
+++ b/ppgan/datasets/preprocess/transforms.py
@@ -41,6 +41,7 @@ TRANSFORMS.register(T.RandomHorizontalFlip)
 TRANSFORMS.register(T.RandomVerticalFlip)
 TRANSFORMS.register(T.Normalize)
 TRANSFORMS.register(T.Transpose)
+TRANSFORMS.register(T.Grayscale)
 
 
 @PREPROCESS.register()
diff --git a/setup.py b/setup.py
index ff5d222dcb7c7b126d5135680068a8285a9f7ede..44c31d1e8cc658681d1950eeeefb6a79b4d54527 100644
--- a/setup.py
+++ b/setup.py
@@ -13,6 +13,7 @@
 # limitations under the License.
 
 from setuptools import setup
+from setuptools import find_packages
 from io import open
 
 with open('requirements.txt', encoding="utf-8-sig") as f:
@@ -27,11 +28,11 @@ def readme():
 
 setup(
     name='ppgan',
-    packages=['ppgan'],
+    packages=find_packages(),
     include_package_data=True,
     entry_points={"console_scripts": ["paddlegan= paddlegan.paddlegan:main"]},
     author='PaddlePaddle Author',
-    version='0.1.0',
+    version='2.0.0-beta',
     install_requires=requirements,
     license='Apache License 2.0',
     description='Awesome GAN toolkits based on PaddlePaddle',