Skip to content

Commit

Permalink
fix warning message (#1157)
Browse files Browse the repository at this point in the history
(cherry picked from commit eed1f0e)
  • Loading branch information
JamesLYC88 authored and Borda committed Jul 22, 2022
1 parent 6bb5542 commit 909ad97
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torchmetrics/metric.py
Expand Up @@ -143,7 +143,7 @@ def __init__(
default needs access to the full metric state. If this is not the case, significant speedups can be
achieved and we recommend setting this to `False`.
We provide an checking function
`from torchmetrics.utilities import check_forward_no_full_state`
`from torchmetrics.utilities import check_forward_full_state_property`
that can be used to check if the `full_state_update=True` (old and potential slower behaviour,
default for now) or if `full_state_update=False` can be used safely.
""",
Expand Down

0 comments on commit 909ad97

Please sign in to comment.