mostly new containers

DJCLosses.py

@@ -6,6 +6,6 @@
 def dummy_loss(truth, pred):
     #t = tf.Print(truth,[truth],'truth ')
     #p = tf.Print(pred,[pred],'pred ')
-    return tf.reduce_mean(truth)+tf.reduce_mean(pred)
+    return (tf.reduce_mean(truth)-tf.reduce_mean(pred))**2
 
 djc_global_loss_list['dummy_loss']=dummy_loss

docker/Dockerfile_base

@@ -64,11 +64,10 @@ RUN pip3 install gpustat setGPU
 
 ## torch/jupyter stuff 
 RUN pip3 install future gensim jupyter prompt-toolkit 
-RUN export CUDA="cu101" && \
-    pip3 install torch-scatter==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html  && \
-    pip3 install torch-sparse==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html  && \
-    pip3 install torch-cluster==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html  && \
-    pip3 install torch-spline-conv==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html  && \
+RUN pip3 install torch-scatter==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.6.0.html  && \
+    pip3 install torch-sparse==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.6.0.html  && \
+    pip3 install torch-cluster==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.6.0.html  && \
+    pip3 install torch-spline-conv==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.6.0.html  && \
     pip3 install torch-geometric
 
 
@@ -78,6 +77,10 @@ EXPOSE 8888
 RUN adduser  --disabled-password --gecos ""  dummyuser
 RUN apt-get install -y vim
 
+#some tex things
+RUN apt install -y texlive ghostscript dvipng
+
+
 
 
 

docker/Dockerfile_exp

@@ -1,13 +1,13 @@
 
-FROM cernml4reco/djcbase:py3
+FROM cernml4reco/djcbase:latest
 
 
 
 ################################################################################
 # Tensorflow
 
 RUN export DEBIAN_FRONTEND=noninteractive && \
-    pip3 --no-cache-dir install tensorflow-gpu==2.2.0 
+    pip3 --no-cache-dir install tensorflow-gpu==2.3.0
 
 
 #2.2.0-rc2