From 625284a2e62859f60c4abc3a89fc178594109cfb Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Thu, 4 Feb 2021 17:40:44 +0100
Subject: [PATCH] Log device.
---
pysegcnn/core/trainer.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py
index 271633e..0cf2da5 100644
--- a/pysegcnn/core/trainer.py
+++ b/pysegcnn/core/trainer.py
@@ -2483,6 +2483,7 @@ class NetworkInference(BaseConfig):
LOGGER.info('Setting model to evaluation mode ...')
model.eval()
model.to(self.device)
+ LOGGER.info('Device: {}'.format(self.device))
# iterate over the samples of the target dataset
output = {}
--
GitLab