From 7e5eaa049aac0ba7e3be53b43bd8c64eb7ce4aa1 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Thu, 4 Feb 2021 17:48:01 +0100
Subject: [PATCH] Fixed a bug: Join argument needs to be iterable.

---
 pysegcnn/core/trainer.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py
index 0cf2da5..be12dd3 100644
--- a/pysegcnn/core/trainer.py
+++ b/pysegcnn/core/trainer.py
@@ -2526,8 +2526,8 @@ class NetworkInference(BaseConfig):
             output[batch] = {'x': inputs, 'y': labels, 'y_pred': prdctn}
 
             # filename for the plot of the current batch
-            batch_name = '_'.join(model.state_file.stem,
-                                  '{}_{}.pt'.format(self.trg_ds.name, batch))
+            batch_name = '_'.join([model.state_file.stem,
+                                  '{}_{}.pt'.format(self.trg_ds.name, batch)])
 
             # check if the current batch name exceeds the Windows limit of
             # 255 characters
-- 
GitLab