train on 3 epochs for pipe mode test (#64)

Author: andremoeller

test/resources/synthetic/synthetic_pipe_mode_dataset.py

@@ -7,7 +7,7 @@
 BATCH_SIZE = 64
 NUM_PARALLEL_BATCHES = 2
 DIMENSION = 1024
-EPOCHS = 1
+EPOCHS = 3
 
 def estimator_fn(run_config, params):
     column = tf.feature_column.numeric_column('data', shape=(DIMENSION, ))
@@ -41,4 +41,4 @@ def parse(record):
     ds = ds.apply(map_and_batch(parse, batch_size=BATCH_SIZE,
                                 num_parallel_batches=NUM_PARALLEL_BATCHES))
 
-    return ds
+    return ds