detach result from graph

mohamad-hasan-sohan-ajini · Jul 9, 2020 · ace014f · ace014f
1 parent 38f5745
commit ace014f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/evaluation.py b/evaluation.py
@@ -53,7 +53,7 @@ def utternace_repr(model, transform, num_frames, device, filepath):
         x = model(x)
         x = x.mean(dim=0)
 
-    return x
+    return x.detach()
 
 
 def cosine_based(rep0, rep1):