Merge pull request PaddlePaddle#34 from heavengate/update_paddle_io

update paddle.io
qingqing01 · Apr 14, 2020 · d8541ea · d8541ea
2 parents 4bf3662 + 84362df
commit d8541ea
Show file tree

Hide file tree

Showing 82 changed files with 805 additions and 3,617 deletions.
diff --git a/.gitignore b/.gitignore
@@ -2,3 +2,6 @@
 *.json
 output*
 *checkpoint*
+build
+dist
+hapi.egg-info
diff --git a/callbacks.py b/callbacks.py
diff --git a/cyclegan/data.py b/cyclegan/data.py
@@ -30,7 +30,7 @@
 import paddle.fluid as fluid
 
 
-class Cityscapes(fluid.io.Dataset):
+class Cityscapes(paddle.io.Dataset):
     def __init__(self, root_path, file_path, mode='train', return_name=False):
         self.root_path = root_path
         self.file_path = file_path

diff --git a/cyclegan/train.py b/cyclegan/train.py
@@ -86,13 +86,13 @@ def main():
     if FLAGS.resume:
         g.load(FLAGS.resume)
 
-    loader_A = fluid.io.DataLoader(
+    loader_A = paddle.io.DataLoader(
         data.DataA(),
         places=place,
         shuffle=True,
         return_list=True,
         batch_size=FLAGS.batch_size)
-    loader_B = fluid.io.DataLoader(
+    loader_B = paddle.io.DataLoader(
         data.DataB(),
         places=place,
         shuffle=True,
-Original file line number
+Diff line change
@@ Expand Up / @@ -2,3 +2,6 @@ @@
     *.json
     output*
     *checkpoint*
+    build
+    dist
+    hapi.egg-info