From 3ef9030fdf3ce40bc09c376bd2bc688488aa347b Mon Sep 17 00:00:00 2001
From: sdtblck <sdtblck@gmail.com>
Date: Tue, 16 Jun 2020 21:59:49 +0100
Subject: [PATCH] -
---
training/training_loop.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/training/training_loop.py b/training/training_loop.py
index 1895eaa..e78491a 100755
--- a/training/training_loop.py
+++ b/training/training_loop.py
@@ -39,8 +39,8 @@ def _random_choice(inputs, n_samples=1):
def apply_random_aug(x):
with tf.name_scope('SpatialAugmentations'):
choice = tf.random_uniform([], 0, 2, tf.int32)
- x = tf.cond(tf.reduce_all(tf.equal(choice, tf.constant(0))), misc.zoom_in(x), lambda: tf.identity(x))
- x = tf.cond(tf.reduce_all(tf.equal(choice, tf.constant(1))), misc.zoom_in(x), lambda: tf.identity(x))
+ x = tf.cond(tf.reduce_all(tf.equal(choice, tf.constant(0))), lambda: misc.zoom_in(x), lambda: tf.identity(x))
+ x = tf.cond(tf.reduce_all(tf.equal(choice, tf.constant(1))), lambda: misc.zoom_in(x), lambda: tf.identity(x))
# if :
# print('zooming in')
# x = misc.zoom_in(x)
--
GitLab