Skip to content

Commit c6c150b

Browse files
committed
Update SiameseHead
We forgot to pass the train parameter to the layer.__call__ inside get_output_at. This is a bug.
1 parent ff676f1 commit c6c150b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

keras/layers/core.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1743,7 +1743,7 @@ def set_layer_input(self, head):
17431743
def get_output_at(self, head, train=False):
17441744
X = self.inputs[head].get_output(train)
17451745
mask = self.inputs[head].get_output_mask(train)
1746-
Y = self.layer(X, mask)
1746+
Y = self.layer(X, mask=mask, train=train)
17471747
return Y
17481748

17491749
def get_output_shape(self, head, train=False):

0 commit comments

Comments
 (0)