Skip to content

Commit 8f67bab

Browse files
authored
Update layer.py add_loss to properly handle when input loss is a list of scalar tensor (#21291)
1 parent df58e70 commit 8f67bab

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

keras/src/layers/layer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1203,8 +1203,8 @@ def call(self, x):
12031203
scope = backend.get_stateless_scope()
12041204
if scope.collect_losses:
12051205
for x in losses:
1206-
scope.add_loss(loss)
1207-
self._loss_ids.add(id(loss))
1206+
scope.add_loss(x)
1207+
self._loss_ids.add(id(x))
12081208
else:
12091209
self._losses.extend(losses)
12101210

0 commit comments

Comments
 (0)