From 05f0ddfc15e35e3344461a640a2931879214c815 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Thu, 4 Feb 2021 17:51:36 +0100
Subject: [PATCH] Fixed a bug: Read dataset not Subset
---
pysegcnn/core/trainer.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py
index be12dd3..d09a6c3 100644
--- a/pysegcnn/core/trainer.py
+++ b/pysegcnn/core/trainer.py
@@ -2607,7 +2607,7 @@ class NetworkInference(BaseConfig):
domain=self.domain)
# load the source dataset the model was trained on
- self.src_ds = self.load_dataset(state, test=None)
+ self.src_ds = self.load_dataset(state, test=None).dataset
# load the pretrained model
model, _ = Network.load_pretrained_model(state)
--
GitLab