centering export fix

krzynio · Apr 26, 2018 · a620cc8 · a620cc8
1 parent 0546ef8
commit a620cc8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/utils.py b/src/utils.py
@@ -415,8 +415,8 @@ def normalize_embeddings(emb, types, mean=None):
             continue
         if t == 'center':
             if mean is None:
-                mean = emb.mean(0, keepdim=True).expand_as(emb)
-            emb.sub_(mean)
+                mean = emb.mean(0, keepdim=True)
+            emb.sub_(mean.expand_as(emb))
         elif t == 'renorm':
             emb.div_(emb.norm(2, 1, keepdim=True).expand_as(emb))
         else: