Skip to content

Commit 0d2ceb1

Browse files
authored
Merge pull request #2 from vbvg2008/master
added pipeline benchmark example
2 parents 692d6ed + 0b89071 commit 0d2ceb1

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

benchmark/pipeline_performance.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
from fastestimator.pipeline.static.preprocess import Minmax
2+
from fastestimator.pipeline.pipeline import Pipeline
3+
import tensorflow as tf
4+
import numpy as np
5+
6+
(x_train, y_train), (x_eval, y_eval) = tf.keras.datasets.mnist.load_data()
7+
x_train = np.expand_dims(x_train, -1)
8+
9+
pipeline = Pipeline(batch_size=32,
10+
feature_name=["x", "y"],
11+
train_data={"x": x_train, "y": y_train},
12+
validation_data= {"x": x_eval, "y": y_eval},
13+
transform_train= [[Minmax()], []])
14+
15+
pipeline.benchmark(num_steps=3000, mode="train")

0 commit comments

Comments
 (0)