meg HF staff commited on
Commit
a55a160
·
verified ·
1 Parent(s): ff4d1d5

Update train.sh

Browse files
Files changed (1) hide show
  1. train.sh +1 -1
train.sh CHANGED
@@ -8,7 +8,7 @@ ls /app/*
8
  cd teaching_arithmetic
9
  mkdir -p out
10
  python train.py config2/addition/plain/train_addition_bal.py \
11
- --ckpt_path_name="ckpt_10000.pt" \
12
  --out_dir='out/addition_plain' \
13
  --data_type='text' --data_format='plain' \
14
  --dataset='bal' --train_data_path="train_3digit_10000.txt" \
 
8
  cd teaching_arithmetic
9
  mkdir -p out
10
  python train.py config2/addition/plain/train_addition_bal.py \
11
+ --ckpt_path_name="ckpt_10000.pt" --wandb_entity='' \
12
  --out_dir='out/addition_plain' \
13
  --data_type='text' --data_format='plain' \
14
  --dataset='bal' --train_data_path="train_3digit_10000.txt" \