diff --git a/mlair/model_modules/model_class.py b/mlair/model_modules/model_class.py
index 4b507db3b2590dcf043f9aa8ed23935115adceef..04293c8f5c0ecf73e7d51fe6ae97e8ff7db96252 100644
--- a/mlair/model_modules/model_class.py
+++ b/mlair/model_modules/model_class.py
@@ -121,6 +121,8 @@ __author__ = "Lukas Leufen, Felix Kleinert"
 __date__ = '2020-05-12'
 
 import tensorflow.keras as keras
+import tensorflow as tf
+import numpy as np
 
 from mlair.model_modules import AbstractModelClass
 from mlair.model_modules.inception_model import InceptionModelBase
@@ -610,6 +612,8 @@ class MyUnet(AbstractModelClass):
 
         self.dense_units = 32*2
         self.initial_lr = 0.001
+        print(f"tf.__version__ = {tf.__version__}")
+        print(f"np.__version__ = {np.__version__}")
 
         # apply to model
         self.set_model()