Plot model state loss.

climax/core/utils.py

@@ -119,8 +119,7 @@ def plot_loss(state_file, figsize=(10, 10), step=5):
     Parameters
     ----------
     state_file : `str` or :py:class:`pathlib.Path`
-        The model state file. Model state files are stored in
-        `pysegcnn/main/_models`.
+        The model state file.
     figsize : `tuple` [`int`], optional
         The figure size in centimeters. The default is `(10, 10)`.
     step : `int`, optional