Skip to content
Snippets Groups Projects
Commit 308de19f authored by Felix Kleinert's avatar Felix Kleinert
Browse files

print statements for version testing in model class

parent 511bd425
Branches
Tags
No related merge requests found
Pipeline #82633 failed
......@@ -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()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment