update loss

yaodongyu · Jan 29, 2019 · 164c135 · 164c135
1 parent 76f4044
commit 164c135
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/trades.py b/trades.py
@@ -69,3 +69,4 @@ def trades_loss(model,
                                                     F.softmax(model(x_natural), dim=1))
     loss = loss_natural + beta * loss_robust
     return loss
+