diff --git a/paddle_fl/core/trainer/fl_trainer.py b/paddle_fl/core/trainer/fl_trainer.py
index 8c9e621b07a0c7e10e909e3f973bc1fe7a5638c2..28a058bb919bf1ffd622d2d901ee634a5c6a5b16 100755
--- a/paddle_fl/core/trainer/fl_trainer.py
+++ b/paddle_fl/core/trainer/fl_trainer.py
@@ -111,6 +111,20 @@ class FedAvgTrainer(FLTrainer):
     def reset(self):
         self.cur_step = 0
 
+    def run_with_epoch(self,reader,feeder,fetch,num_epoch):
+	self._logger.debug("begin to run recv program")
+        self.exe.run(self._recv_program)
+	epoch = 0
+	for i in range(num_epoch):
+	    print(epoch)
+	    for data in reader():
+		self.exe.run(self._main_program,
+                           feed=feeder.feed(data),
+                           fetch_list=fetch)
+	    self.cur_step += 1
+	    epoch += 1
+	self._logger.debug("begin to run send program")
+        self.exe.run(self._send_program)
     def run(self, feed, fetch):
         self._logger.debug("begin to run FedAvgTrainer, cur_step=%d, inner_step=%d" %
                            (self.cur_step, self._step))
diff --git a/paddle_fl/version.py b/paddle_fl/version.py
index 59449eaf8bf7537eef67ddf17d2fd27da2495593..ba528598b5853e154128d1192e63443980140ed7 100644
--- a/paddle_fl/version.py
+++ b/paddle_fl/version.py
@@ -12,5 +12,5 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """ PaddleFL version string """
-fl_version = "0.1.1"
-module_proto_version = "0.1.1"
+fl_version = "0.1.2"
+module_proto_version = "0.1.2"