From 205e4a9c555ac3935bd9fc4bbf020867d2d02aea Mon Sep 17 00:00:00 2001
From: Andreas Wilke <wilke@anl.gov>
Date: Wed, 19 Oct 2022 12:22:12 -0500
Subject: [PATCH 1/3] Added local Dockerfile; testing version of train.sh

---
 Dockerfile            | 10 ++++++++++
 Pilot1/NT3/Dockerfile | 19 +++++++++++++++++++
 2 files changed, 29 insertions(+)
 create mode 100644 Dockerfile
 create mode 100644 Pilot1/NT3/Dockerfile

diff --git a/Dockerfile b/Dockerfile
new file mode 100644
index 00000000..36c48d97
--- /dev/null
+++ b/Dockerfile
@@ -0,0 +1,10 @@
+FROM ubuntu:22.04
+RUN apt update -y 
+RUN apt install -y git \
+    python3 \
+    python3-pip
+WORKDIR /usr/repos
+RUN pip3 install torch tensorflow-gpu
+RUN git clone https://github.com/ECP-CANDLE/Benchmarks && cd Benchmarks && git checkout develop
+RUN git clone https://github.com/ECP-CANDLE/candle_lib.git && cd candle_lib && git checkout develop && python3 setup.py install
+ENV CANDLE_DATA_DIR=/data/
diff --git a/Pilot1/NT3/Dockerfile b/Pilot1/NT3/Dockerfile
new file mode 100644
index 00000000..fe4a1784
--- /dev/null
+++ b/Pilot1/NT3/Dockerfile
@@ -0,0 +1,19 @@
+FROM jdacs4c/improve-pytorch:0.0.1
+# FROM ubuntu:22.04 
+RUN apt update -y 
+# RUN apt install -y git \
+    # python3 \
+    # python3-pip
+WORKDIR /
+# RUN pip3 install torch tensorflow-gpu
+RUN pip3 install tensorflow-gpu
+
+RUN git clone https://github.com/ECP-CANDLE/Benchmarks && cd Benchmarks && git checkout develop
+RUN git clone https://github.com/ECP-CANDLE/candle_lib.git && cd candle_lib && git checkout develop && python3 setup.py install
+ENV CANDLE_DATA_DIR=/data/
+ENV REPO_DIR=/Benchmarks
+# ASSUMING BUILD IS SPECIFIC BENCHMARK DIRECTORY, e.g. Pilot1/NT3
+COPY *.sh /usr/local/bin/
+
+
+

From 7bd3976ed101cbc418dc464bafff2f9bab0ea498 Mon Sep 17 00:00:00 2001
From: Andreas Wilke <wilke@anl.gov>
Date: Wed, 19 Oct 2022 12:24:35 -0500
Subject: [PATCH 2/3] Added path and options to output

---
 Pilot1/NT3/train.sh | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/Pilot1/NT3/train.sh b/Pilot1/NT3/train.sh
index a7a356d7..13fe7b12 100755
--- a/Pilot1/NT3/train.sh
+++ b/Pilot1/NT3/train.sh
@@ -2,3 +2,12 @@
 
 # TRAIN SH
 # IMPROVE interface to NT3 model
+
+MODEL=nt3_baseline_keras2.py
+MODEL_PATH=`find ${REPO_DIR} -name $MODEL`
+
+echo Train NT3 at ${MODEL_PATH}
+echo Options: $@
+
+python $MODEL_PATH $@
+

From 1eb737163d09b48cf6df0ce4576a5a7d4b679609 Mon Sep 17 00:00:00 2001
From: Andreas Wilke <wilke@anl.gov>
Date: Wed, 19 Oct 2022 12:25:26 -0500
Subject: [PATCH 3/3] Deleted high level Dockerfile

---
 Dockerfile | 10 ----------
 1 file changed, 10 deletions(-)
 delete mode 100644 Dockerfile

diff --git a/Dockerfile b/Dockerfile
deleted file mode 100644
index 36c48d97..00000000
--- a/Dockerfile
+++ /dev/null
@@ -1,10 +0,0 @@
-FROM ubuntu:22.04
-RUN apt update -y 
-RUN apt install -y git \
-    python3 \
-    python3-pip
-WORKDIR /usr/repos
-RUN pip3 install torch tensorflow-gpu
-RUN git clone https://github.com/ECP-CANDLE/Benchmarks && cd Benchmarks && git checkout develop
-RUN git clone https://github.com/ECP-CANDLE/candle_lib.git && cd candle_lib && git checkout develop && python3 setup.py install
-ENV CANDLE_DATA_DIR=/data/