Skip to content

GitLab

Unverified Commit 1d878a12 authored by Gustaf Ahdritz's avatar Gustaf Ahdritz Committed by GitHub
Browse files

Fix assert bug

parent 7fdb503e
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
train_openfold.py
View file @ 1d878a12
...@@ -88,7 +88,6 @@ class OpenFoldWrapper(pl.LightningModule): ...@@ -88,7 +88,6 @@ class OpenFoldWrapper(pl.LightningModule):
) )
for k,v in other_metrics.items(): for k,v in other_metrics.items():
assert(len(v.shape) == 1)
self.log( self.log(
f"{phase}/{k}", f"{phase}/{k}",
torch.mean(v), torch.mean(v),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment