From eee54b7c4e9254e0fa0227b4f889534be8ef1a8e Mon Sep 17 00:00:00 2001
From: rtaiello <41542771+rtaiello@users.noreply.github.com>
Date: Tue, 29 Jun 2021 11:55:06 +0200
Subject: [PATCH] Fix CIFAR10 test path

---
 problems.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/problems.py b/problems.py
index 553556f..481d052 100644
--- a/problems.py
+++ b/problems.py
@@ -213,7 +213,7 @@ def cifar10(path,  # pylint: disable=invalid-name
                               "data_batch_{}.bin".format(i))
                  for i in xrange(1, 6)]
   elif mode == "test":
-    filenames = [os.path.join(path, "test_batch.bin")]
+    filenames = [os.path.join(path, "cifar-10-batches-bin/test_batch.bin")]
   else:
     raise ValueError("Mode {} not recognised".format(mode))
 
@@ -258,7 +258,7 @@ def _conv_activation(x):  # pylint: disable=invalid-name
                             use_batch_norm=batch_norm)
 
   if batch_norm:
-    linear_activation = lambda x: tf.nn.relu(snt.BatchNorm()(x))
+    linear_activation = lambda x: tf.nn.relu(snt.BatchNorm()(x, is_training=True))
   else:
     linear_activation = tf.nn.relu