modified: dsm/utilities.py

autonlab · Feb 13, 2021 · e759e6c · e759e6c
1 parent 6af6ac9
commit e759e6c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dsm/utilities.py b/dsm/utilities.py
@@ -73,7 +73,7 @@ def pretrain_dsm(model, t_train, e_train, t_valid, e_valid,
     optimizer.step()
 
     valid_loss = 0
-    for r in range(model.risks): 
+    for r in range(model.risks):
       valid_loss += unconditional_loss(premodel, t_valid, e_valid, str(r+1))
     valid_loss = valid_loss.detach().cpu().numpy()
     costs.append(valid_loss)