fix backward

Relja · Mar 4, 2016 · e592367 · e592367
1 parent e7b6c3f
commit e592367
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -1,6 +1,6 @@
 # NetVLAD: CNN architecture for weakly supervised place recognition
 
-> Version 1.02 (29 Feb 2016)
+> Version 1.03 (04 Mar 2016)
 >
 >- If you used NetVLAD v1.01 or below, you need to upgrade your models using `relja_simplenn_tidy`
 
@@ -155,6 +155,9 @@ More information is available `README_more.md` and in comments in the code itsel
 
 # Changes
 
+- **1.03** (04 Mar 2016)
+    - Fixed a bug in NetVLAD backprop
+
 - **1.02** (29 Feb 2016)
     - Adapts the code to account for major changes in matconvnet-1.0-beta17's SimpleNN
     - Removed the use of the redundant `relja_simplenn` since `vl_simplenn` has sufficient functionality now (from matconvnet-1.0-beta18)

diff --git a/layerVLAD.m b/layerVLAD.m
@@ -123,8 +123,7 @@
                 repmat( ...
                     reshape( sum(sum(assgn,1),2), [l.K, 1, batchSize] ), ...
                     [1, l.D, 1] ), ...
-                3 ), [1, 1, l.D, l.K] );
-
+                3 )', [1, 1, l.D, l.K] );
         end
 
         function objStruct= saveobj(obj)

diff --git a/layerVLADv2.m b/layerVLADv2.m
@@ -126,7 +126,7 @@
                 repmat( ...
                     reshape( sum(sum(assgn,1),2), [l.K, 1, batchSize] ), ...
                     [1, l.D, 1] ), ...
-                3 ), [1, 1, l.D, l.K] );
+                3 )', [1, 1, l.D, l.K] );
 
         end