Merge pull request lightaime#2 from JoshZastrow/patch-1

Update cnn.py -- Regularization function - fixed a typo
congfu · May 7, 2017 · ebeb3ef · ebeb3ef
2 parents facb902 + 47cc97c
commit ebeb3ef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/assignment2/cs231n/classifiers/cnn.py b/assignment2/cs231n/classifiers/cnn.py
@@ -108,7 +108,7 @@ def loss(self, X, y=None):
     loss, dout = softmax_loss(scores, y)
 
     # Add regularization
-    loss += self.reg * 0.5 * (np.sum(self.params['W1'] ** 2) + np.sum(self.params['W2'] ** 2) + np.sum(self.params['W1'] ** 2))
+    loss += self.reg * 0.5 * (np.sum(self.params['W1'] ** 2) + np.sum(self.params['W2'] ** 2) + np.sum(self.params['W3'] ** 2))
 
     dX3, grads['W3'], grads['b3'] = affine_backward(dout, cache_forward_3)
     dX2 = relu_backward(dX3, cache_relu_2)