From f3ef017ff46525d5079bdf2df8e2fde02a13d810 Mon Sep 17 00:00:00 2001 From: Jeremy Hummel Date: Wed, 14 Aug 2019 11:06:38 -0700 Subject: [PATCH] debug --- models/Model_SAE/Model.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/models/Model_SAE/Model.py b/models/Model_SAE/Model.py index b434f71..4044ee9 100644 --- a/models/Model_SAE/Model.py +++ b/models/Model_SAE/Model.py @@ -436,7 +436,8 @@ class SAEModel(ModelBase): else: if self.options['learn_mask']: - self.AE_convert = K.function([warped_dst], [pred_src_dst[-1], pred_dst_dstm[-1], pred_src_dstm[-1]]) + from keras import K + self.AE_convert = K.function([warped_dst], [pred_src_dst[-1][:4], pred_dst_dstm[-1], pred_src_dstm[-1]]) else: self.AE_convert = K.function([warped_dst], [pred_src_dst[-1]])