Skip to content

Commit

Permalink
make docstring single-line
Browse files Browse the repository at this point in the history
  • Loading branch information
mcw92 committed Mar 12, 2024
1 parent 5dc62bc commit 537b848
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions tutorials/torch_example.py
Original file line number Diff line number Diff line change
Expand Up @@ -164,9 +164,7 @@ def configure_optimizers(self) -> torch.optim.SGD:
return torch.optim.SGD(self.parameters(), lr=self.lr)

def on_validation_epoch_end(self):
"""
Calculate and store the model's validation accuracy after each epoch.
"""
"""Calculate and store the model's validation accuracy after each epoch."""
val_acc_val = self.val_acc.compute()
self.val_acc.reset()
if val_acc_val > self.best_accuracy:
Expand Down

0 comments on commit 537b848

Please sign in to comment.