Uncommenting train.py to try doing it end to end.

train.sh

@@ -1,16 +1,15 @@
 #!/bin/bash
 
 echo "Starting attempt at training with default from git repo!"
-#cd teaching_arithmetic
 ls /app/*
 cd teaching_arithmetic
 mkdir -p out
-#python train.py config2/addition/plain/train_addition_bal.py \
-#--ckpt_path_name="ckpt_10000.pt" \
-#--out_dir='out/addition_plain' --wandb_log=False \
-#--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'
+python train.py config2/addition/plain/train_addition_bal.py \
+--ckpt_path_name="ckpt_10000.pt" \
+--out_dir='out/addition_plain' --wandb_log=False \
+--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 training! Uploading!"
 cd ..