Merge pull request #225 from pytti-tools/not_inplace

[BUGFIX] remove inplace operations from spherical_dist to resolve new backprop error
pytti-tools · Dec 15, 2022 · 9e85683 · 9e85683
2 parents bdd0d96 + 8a5e5c6
commit 9e85683
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/pytti/Perceptor/Prompt.py b/src/pytti/Perceptor/Prompt.py
@@ -34,7 +34,7 @@
 def spherical_dist_loss(x, y):
     x = F.normalize(x, dim=-1)
     y = F.normalize(y, dim=-1)
-    return x.sub_(y).norm(dim=-1).div_(2).arcsin_().pow_(2).mul_(2)
+    return x.sub(y).norm(dim=-1).div(2).arcsin().pow(2).mul(2)
 
 
 def make_mask(mask, thresh):