PeteBleackley commited on
Commit
1d1a876
·
1 Parent(s): 684c1d8

Use builtin sum rather than torch.sum for a generator

Browse files
Files changed (1) hide show
  1. scripts.py +4 -4
scripts.py CHANGED
@@ -33,10 +33,10 @@ class CombinedLoss(torch.nn.Module):
33
  torch.nn.MSELoss())
34
 
35
  def forward(self,y_pred,y_true):
36
- return torch.sum((fn(pred,obs)
37
- for (fn,pred,obs) in zip(self.component_losses,
38
- y_pred,
39
- y_true)))
40
 
41
 
42
  def capitalise(token,i):
 
33
  torch.nn.MSELoss())
34
 
35
  def forward(self,y_pred,y_true):
36
+ return sum((fn(pred,obs)
37
+ for (fn,pred,obs) in zip(self.component_losses,
38
+ y_pred,
39
+ y_true)))
40
 
41
 
42
  def capitalise(token,i):