Skip to content

Commit

Permalink
format code
Browse files Browse the repository at this point in the history
  • Loading branch information
Kaiyu Yang committed Apr 27, 2023
1 parent a8caec7 commit 59396ff
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion verifier/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -59,7 +59,7 @@ def __init__(
self.add_module(f"{name}_{split}", m)

def log_metrics(self, split: str, logit: torch.Tensor, label: torch.Tensor) -> None:
for (name, metric) in self.metrics[split].items():
for name, metric in self.metrics[split].items():
metric(logit, label)
self.log(f"{name}_{split}", metric, on_step=False, on_epoch=True)

Expand Down

0 comments on commit 59396ff

Please sign in to comment.