Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #!/usr/bin/env python3
- import tensorflow as tf
- import tensorflow.contrib.eager as tfe
- import numpy as np
- import time
- tfe.enable_eager_execution()
- while True:
- start = time.time()
- conv_layer = tf.layers.Conv2D(filters=5, kernel_size=3)
- random_input = np.random.normal(size=(1, 10, 10, 3)).astype(np.float32)
- _ = conv_layer(tf.convert_to_tensor(random_input))
- print(time.time() - start)
Add Comment
Please, Sign In to add comment