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