Fix validation loss computation

autonlab · Jan 2, 2021 · 3e991e4 · 3e991e4
1 parent af1871b
commit 3e991e4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/dsm/utilities.py b/dsm/utilities.py
@@ -174,7 +174,8 @@ def train_dsm(model,
                                        x_valid,
                                        t_valid_,
                                        e_valid_,
-                                       elbo=False)
+                                       elbo=False,
+                                       risk=str(r+1))
 
     valid_loss = valid_loss.detach().cpu().numpy()
     costs.append(float(valid_loss))