Skip to content

Commit d20fe64

Browse files
committed
Merge pull request keras-team#1823 from EderSantana/patch-6
Update SiameseHead
2 parents ababd95 + c6c150b commit d20fe64

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
@@ -1744,7 +1744,7 @@ def set_layer_input(self, head):
17441744
def get_output_at(self, head, train=False):
17451745
X = self.inputs[head].get_output(train)
17461746
mask = self.inputs[head].get_output_mask(train)
1747-
Y = self.layer(X, mask)
1747+
Y = self.layer(X, mask=mask, train=train)
17481748
return Y
17491749

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

0 commit comments

Comments
 (0)