From cea5d86d3df8a5c2fc34b42fe3eb7cddb5ea2168 Mon Sep 17 00:00:00 2001 From: lvapeab Date: Thu, 1 Jun 2017 10:36:29 +0200 Subject: [PATCH] Use default attention size if unspecified --- model_zoo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model_zoo.py b/model_zoo.py index a2e07eb..f54ae3d 100644 --- 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']),