Skip to content

Commit cea5d86

Browse files
committed
Use default attention size if unspecified
1 parent e5ada24 commit cea5d86

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

model_zoo.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -351,7 +351,7 @@ def GroundHogModel(self, params):
351351
initial_memory = None
352352
# 3.3. Attentional decoder
353353
sharedAttRNNCond = eval('Att' + params['RNN_TYPE'] + 'Cond')(params['DECODER_HIDDEN_SIZE'],
354-
att_dim=params['ATTENTION_SIZE'],
354+
att_dim=params.get('ATTENTION_SIZE', 0),
355355
W_regularizer=l2(params['RECURRENT_WEIGHT_DECAY']),
356356
U_regularizer=l2(params['RECURRENT_WEIGHT_DECAY']),
357357
V_regularizer=l2(params['RECURRENT_WEIGHT_DECAY']),

0 commit comments

Comments
 (0)