diff --git a/climax/core/utils.py b/climax/core/utils.py index 26007b8607306010f2e75b377b483bffeae82648..cc99955f6ab51cd570d1c342aab53a9b08332217 100644 --- a/climax/core/utils.py +++ b/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