Fix assert bug

aqlaboratory · Jul 7, 2023 · 1d878a1 · 1d878a1
1 parent 7fdb503
commit 1d878a1
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/train_openfold.py b/train_openfold.py
@@ -88,7 +88,6 @@ def _log(self, loss_breakdown, batch, outputs, train=True):
             )
 
         for k,v in other_metrics.items():
-            assert(len(v.shape) == 1)
             self.log(
                 f"{phase}/{k}",
                 torch.mean(v),