diff --git a/pysegcnn/core/graphics.py b/pysegcnn/core/graphics.py
index 199946c4b71b078d72517531068bf8a5ce598dca..82badfc54bbdb776a2be85ff4d56ae6e7664f6d1 100644
--- a/pysegcnn/core/graphics.py
+++ b/pysegcnn/core/graphics.py
@@ -479,7 +479,9 @@ def plot_loss(state_file, figsize=(10, 10), step=5,
 
     """
     # load the model state
+    state_file = pathlib.Path(state_file)
     model_state = torch.load(state_file)
+    LOGGER.info('Plot model statistics: {}'.format(state_file.stem))
 
     # get all non-zero elements, i.e. get number of epochs trained before
     # early stop