From 164c135b95632bf655a14157e4724f14010c40ff Mon Sep 17 00:00:00 2001
From: Yaodong Yu <yaodong.yu92@gmail.com>
Date: Mon, 28 Jan 2019 20:14:09 -0500
Subject: [PATCH] update loss

---
 trades.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/trades.py b/trades.py
index 334cf0a..58ac272 100644
--- 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
+