Use default attention size if unspecified

lvapeab · Jun 1, 2017 · cea5d86 · cea5d86
1 parent e5ada24
commit cea5d86
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model_zoo.py b/model_zoo.py
@@ -351,7 +351,7 @@ def GroundHogModel(self, params):
             initial_memory = None
         # 3.3. Attentional decoder
         sharedAttRNNCond = eval('Att' + params['RNN_TYPE'] + 'Cond')(params['DECODER_HIDDEN_SIZE'],
-                                                                     att_dim=params['ATTENTION_SIZE'],
+                                                                     att_dim=params.get('ATTENTION_SIZE', 0),
                                                                      W_regularizer=l2(params['RECURRENT_WEIGHT_DECAY']),
                                                                      U_regularizer=l2(params['RECURRENT_WEIGHT_DECAY']),
                                                                      V_regularizer=l2(params['RECURRENT_WEIGHT_DECAY']),