fix some typos

geoopt · May 16, 2019 · f10f08d · f10f08d
1 parent d191594
commit f10f08d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/geoopt/samplers/rsgld.py b/geoopt/samplers/rsgld.py
@@ -51,7 +51,7 @@ def step(self, closure):
                     n = torch.randn_like(p).mul_(math.sqrt(epsilon))
                     r = egrad2rgrad(p, 0.5 * epsilon * p.grad + n)
                     # use copy only for user facing point
-                    p.copy_(retr(p, r, 1.0))
+                    copy_or_set(p, retr(p, r, 1.0))
                     p.grad.zero_()
 
         if not self.burnin:

diff --git a/geoopt/tensor.py b/geoopt/tensor.py
@@ -1,6 +1,7 @@
 import torch.nn
 from .manifolds import Euclidean
 from .docutils import insert_docs
+from .utils import copy_or_set
 
 __all__ = ["ManifoldTensor", "ManifoldParameter"]
 
@@ -37,7 +38,7 @@ def proj_(self):
         tensor
             same instance
         """
-        return self.copy_(self.manifold.projx(self))
+        return copy_or_set(self, self.manifold.projx(self))
 
     @insert_docs(Euclidean.retr.__doc__, r"\s+x : .+\n.+", "")
     def retr(self, u, t=1.0, order=None):