diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py
index 271633e43fe979a9cae7c09fa7e1b6107af9bea0..0cf2da50d540749615e7147bae5fd5bf158ad8e3 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 = {}