train.sh

#!/bin/bash

wandb login ${wandb_api_key}
echo "Starting attempt at training with default from git repo!"
#cd teaching_arithmetic
# --wandb_log=False \
ls /app/*
cd teaching_arithmetic
mkdir -p out
python train.py config2/addition/plain/train_addition_bal.py \
--ckpt_path_name="ckpt_10000.pt" --wandb_entity='' \
--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 training! Uploading!"
cd ..
python upload_results.py
echo "Done?"
python pause_space.py