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/ + + + 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 $@ +