From cf8f14ef7364df4a3d1741ecf9425db20ef3c4da Mon Sep 17 00:00:00 2001
From: Nikolai Hartmann <Nikolai.Hartmann@physik.uni-muenchen.de>
Date: Fri, 17 Aug 2018 15:43:39 +0200
Subject: [PATCH] fixing skip_activation mode
---
toolkit.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/toolkit.py b/toolkit.py
index 3896daf..7f45570 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -915,9 +915,13 @@ class ClassificationProject(object):
elif mode == "skip_activation":
# output before applying activation function
# (after weighted sum + bias of last hidden layer)
+ if isinstance(self.model.input, list):
+ feed_dict={tuple(self.model.input) : x}
+ else:
+ feed_dict={self.model.input : x}
return K.get_session().run(
self.model.output.op.inputs[0],
- feed_dict={self.model.input : x}
+ feed_dict=feed_dict
)
else:
raise ValueError("Unknown mode {}".format(mode))
--
GitLab