Spaces:
Build error
Build error
PeteBleackley
commited on
Commit
·
1d1a876
1
Parent(s):
684c1d8
Use builtin sum rather than torch.sum for a generator
Browse files- 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
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
|
| 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):
|