python3 migration, will tag as 3.0

Author: jkiesele

docker/Dockerfile_base

@@ -63,6 +63,13 @@ RUN pip3 install opencv-python easydict tqdm Cython numba
 RUN pip3 install gpustat setGPU
 
 ## torch/jupyter stuff 
-RUN pip3 install torch future gensim jupyter prompt-toolkit 
+RUN pip3 install torch future gensim jupyter prompt-toolkit torch_geometric
 EXPOSE 8888
 
+#for singularity panic
+RUN adduser  --disabled-password --gecos ""  dummyuser
+
+
+
+
+

environment/activateROOT.sh

@@ -1,31 +1,6 @@
-import imp
-try:
-    imp.find_module('Losses')
-    from Losses import *
-except ImportError:
-    print 'No Losses module found, ignoring at your own risk'
-    global_loss_list = {}
-
-try:
-    imp.find_module('Layers')
-    from Layers import *
-except ImportError:
-    print 'No Layers module found, ignoring at your own risk'
-    global_layers_list = {}
-
-try:
-    imp.find_module('Metrics')
-    from Metrics import *
-except ImportError:
-    print 'No metrics module found, ignoring at your own risk'
-    global_metrics_list = {}
-
-
-
-custom_objs = {}
-custom_objs.update(global_loss_list)
-custom_objs.update(global_layers_list)
-custom_objs.update(global_metrics_list)
+from DeepJetCore.customObjects import *
+
+custom_objs = get_custom_objects()
 
 
 def getLayer(model, name):
@@ -35,7 +10,6 @@ def getLayer(model, name):
 
 
 
-
 def printLayerInfosAndWeights(model, noweights=False):
     for layer in model.layers:
         g=layer.get_config()