train.sh

#!/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'

echo "Done training! Uploading!"
cd ..
python upload_results.py

echo "Done?"