meg's picture
meg HF staff
Update train.sh
f60da9f verified
raw
history blame
520 Bytes
#!/bin/bash
echo "Starting attempt at training with default from git repo!"
ls *
ls /data/*
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?"