fix typos in scripts

tf/scripts/lm1b_base_gpu.sh

@@ -41,7 +41,7 @@ if [[ $1 == 'train_data' ]]; then
 elif [[ $1 == 'test_data' ]]; then
     python data_utils.py \
       --data_dir=${DATA_ROOT}/ \
-      --dataset=enwik8 \
+      --dataset=lm1b \
       --tgt_len=${TEST_TGT_LEN} \
       --per_host_test_bsz=${TEST_BSZ} \
       --num_passes=1 \

tf/scripts/lm1b_large_tpu.sh

@@ -90,7 +90,7 @@ elif [[ $1 == 'train' ]]; then
         --train_steps=1200000 \
         --tgt_len=${TGT_LEN} \
         --mem_len=${MEM_LEN} \
-        --train_batch_size=${BSZ} \
+        --train_batch_size=${TRAIN_BSZ} \
         --num_hosts=${NUM_HOST} \
         --num_core_per_host=${NUM_CORE} \
         --iterations=1000 \