Update train.sh
Browse files
train.sh
CHANGED
|
@@ -2,12 +2,13 @@
|
|
| 2 |
|
| 3 |
echo "Starting attempt at training with default from git repo!"
|
| 4 |
#cd teaching_arithmetic
|
|
|
|
| 5 |
ls /app/*
|
| 6 |
cd teaching_arithmetic
|
| 7 |
mkdir -p out
|
| 8 |
python train.py config2/addition/plain/train_addition_bal.py \
|
| 9 |
--ckpt_path_name="ckpt_10000.pt" \
|
| 10 |
-
--out_dir='out/addition_plain'
|
| 11 |
--data_type='text' --data_format='plain' \
|
| 12 |
--dataset='bal' --train_data_path="train_3digit_10000.txt" \
|
| 13 |
--eval_addition=True --start='FILE:data/bal/test_10000.txt'
|
|
|
|
| 2 |
|
| 3 |
echo "Starting attempt at training with default from git repo!"
|
| 4 |
#cd teaching_arithmetic
|
| 5 |
+
# --wandb_log=False \
|
| 6 |
ls /app/*
|
| 7 |
cd teaching_arithmetic
|
| 8 |
mkdir -p out
|
| 9 |
python train.py config2/addition/plain/train_addition_bal.py \
|
| 10 |
--ckpt_path_name="ckpt_10000.pt" \
|
| 11 |
+
--out_dir='out/addition_plain' \
|
| 12 |
--data_type='text' --data_format='plain' \
|
| 13 |
--dataset='bal' --train_data_path="train_3digit_10000.txt" \
|
| 14 |
--eval_addition=True --start='FILE:data/bal/test_10000.txt'
|