We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent e5ada24 commit cea5d86Copy full SHA for cea5d86
model_zoo.py
@@ -351,7 +351,7 @@ def GroundHogModel(self, params):
351
initial_memory = None
352
# 3.3. Attentional decoder
353
sharedAttRNNCond = eval('Att' + params['RNN_TYPE'] + 'Cond')(params['DECODER_HIDDEN_SIZE'],
354
- att_dim=params['ATTENTION_SIZE'],
+ att_dim=params.get('ATTENTION_SIZE', 0),
355
W_regularizer=l2(params['RECURRENT_WEIGHT_DECAY']),
356
U_regularizer=l2(params['RECURRENT_WEIGHT_DECAY']),
357
V_regularizer=l2(params['RECURRENT_WEIGHT_DECAY']),
0 commit comments