Adding train script

Dockerfile

@@ -62,7 +62,8 @@ RUN pip install --no-cache-dir --upgrade -r requirements.txt
 #COPY --chown=user teaching_arithmetic /teaching_arithmetic
 #COPY --chown=user . /app
 COPY . /app
-
+COPY ./train.sh /train.sh
 # USER 1000
 
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+# CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+ENTRYPOINT ["/train.sh"]

train.sh

@@ -0,0 +1,11 @@
+#!/bin/bash
+
+echo "Starting attempt at training with default from git repo!"
+python train.py config2/addition/plain/train_addition_bal.py \
+--ckpt_path_name="ckpt_10000.pt" \
+--out_dir='out/addition_plain' \
+--data_type='text' --data_format='plain' \
+--dataset='bal' --train_data_path="train_3digit_10000.txt" \
+--eval_addition=True --start='FILE:data/bal/test_10000.txt'
+
+echo "Done?"