minor fixes in model construction

Author: MartinXPN

layers/QuestionPooling.py

@@ -15,15 +15,15 @@ def __init__(self, **kwargs):
         self.supports_masking = True
 
     def compute_output_shape(self, input_shape):
-        assert(isinstance(input_shape, list) and len(input_shape) == 4)
+        assert(isinstance(input_shape, list) and len(input_shape) == 5)
 
         input_shape = input_shape[0]
         B, Q, H = input_shape
 
         return (B, H)
 
     def build(self, input_shape):
-        assert(isinstance(input_shape, list) and len(input_shape) == 4)
+        assert(isinstance(input_shape, list) and len(input_shape) == 5)
         input_shape = input_shape[0]
 
         B, Q, H_ = input_shape

layers/__init__.py

@@ -5,3 +5,4 @@
 from Argmax import Argmax
 from Slice import Slice
 from SharedWeight import SharedWeightLayer, SharedWeight
+from VariationalDropout import VariationalDropout

model.py

@@ -16,6 +16,7 @@
 from layers import SelfAttnGRU
 from layers import PointerGRU
 from layers import QuestionPooling
+from layers import VariationalDropout
 from layers import Slice
 from layers import SharedWeight