Skip to content

Commit e28d0d7

Browse files
committed
fix loss = ()
1 parent 2822774 commit e28d0d7

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

alf/algorithms/algorithm.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1208,8 +1208,10 @@ def update_with_gradient(self,
12081208

12091209
loss_info = self._aggregate_loss(loss_info, valid_masks, batch_info)
12101210

1211-
all_params, gns = self._backward_and_gradient_update(loss_info.loss *
1212-
weight)
1211+
loss = loss_info.loss
1212+
if loss != ():
1213+
loss = loss * weight
1214+
all_params, gns = self._backward_and_gradient_update(loss)
12131215

12141216
loss_info = loss_info._replace(gns=gns)
12151217
loss_info = alf.nest.map_structure(torch.mean, loss_info)

0 commit comments

Comments
 (0)