update mcov

SMLC-NYSBC · Apr 28, 2023 · 1ce2069 · 1ce2069
1 parent e0e7e6f
commit 1ce2069
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tardis_pytorch/dist_pytorch/trainer.py b/tardis_pytorch/dist_pytorch/trainer.py
@@ -346,7 +346,7 @@ def _validate(self):
                 input0_9 = self.Graph0_9.patch_to_segment(
                     graph=edge_cpu, coord=coord, idx=out_cpu, prune=5, sort=False
                 )
-                mcov_m = mcov(input0_9, target)
+                mcov_m, _ = mcov(input0_9, target)
                 mcov0_9.append(mcov_m)
             except:
                 mcov0_9.append(0.0)