From fff432728748bd44a3589ded0afdac92a8c75567 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Tue, 19 Oct 2021 12:52:52 +0200
Subject: [PATCH] Plot model state loss.

---
 climax/core/utils.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/climax/core/utils.py b/climax/core/utils.py
index 26007b8..cc99955 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
-- 
GitLab