From 308de19f00ad418a5cee57ea56014492a5a566c5 Mon Sep 17 00:00:00 2001
From: Felix Kleinert <f.kleinert@fz-juelich.de>
Date: Tue, 9 Nov 2021 10:18:03 +0100
Subject: [PATCH] print statements for version testing in model class

---
 mlair/model_modules/model_class.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/mlair/model_modules/model_class.py b/mlair/model_modules/model_class.py
index 4b507db3..04293c8f 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()
-- 
GitLab