From 8f87aff8524f200ded2666967eee86d8f7835fca Mon Sep 17 00:00:00 2001 From: Jeremy Hummel Date: Wed, 14 Aug 2019 11:07:55 -0700 Subject: [PATCH] debug --- models/Model_SAE/Model.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/models/Model_SAE/Model.py b/models/Model_SAE/Model.py index 4044ee9..eb6ef79 100644 --- a/models/Model_SAE/Model.py +++ b/models/Model_SAE/Model.py @@ -436,10 +436,9 @@ class SAEModel(ModelBase): else: if self.options['learn_mask']: - 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]]) + self.AE_convert = K.function([warped_dst], [pred_src_dst[-1][:4]]) if self.is_training_mode: self.src_sample_losses = []