Enable grad when doing mst decoding

yzhangcs · yzhangcs · commit c0b678f2994e · 2021-12-27T17:57:48.000+08:00
diff --git a/supar/structs/tree.py b/supar/structs/tree.py
@@ -58,10 +58,10 @@ def __add__(self, other):
 
     @lazy_property
     def argmax(self):
-        with torch.no_grad():
-            return mst(self.scores, self.mask, self.multiroot)
+        return mst(self.scores, self.mask, self.multiroot)
 
     def kmax(self, k):
+        # TODO: Camerini algorithm
         raise NotImplementedError
 
     def sample(self):
