From c50b5d9b3b5754f8bb84a5b290a22b3688c7d30a Mon Sep 17 00:00:00 2001
From: Stefan Schweter <stefan@schweter.it>
Date: Thu, 30 Nov 2017 17:36:19 +0100
Subject: [PATCH] capsnet: fix margin loss formula

---
 extra_capsnets.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/extra_capsnets.ipynb b/extra_capsnets.ipynb
index 67e67bd..89e343e 100644
--- a/extra_capsnets.ipynb
+++ b/extra_capsnets.ipynb
@@ -1191,7 +1191,7 @@
    "source": [
     "The paper uses a special margin loss to make it possible to detect two or more different digits in each image:\n",
     "\n",
-    "$ L_k = T_k \\max(0, m^{+} - \\|\\mathbf{v}_k\\|)^2 - \\lambda (1 - T_k) \\max(0, \\|\\mathbf{v}_k\\| - m^{-})^2$\n",
+    "$ L_k = T_k \\max(0, m^{+} - \\|\\mathbf{v}_k\\|)^2 + \\lambda (1 - T_k) \\max(0, \\|\\mathbf{v}_k\\| - m^{-})^2$\n",
     "\n",
     "* $T_k$ is equal to 1 if the digit of class $k$ is present, or 0 otherwise.\n",
     "* In the paper, $m^{+} = 0.9$, $m^{-} = 0.1$ and $\\lambda = 0.5$.\n",