Skip to content

Commit 3fb2905

Browse files
committed
added bias
1 parent 237d972 commit 3fb2905

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/test_finetuning_cpu.sh

+2
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,7 @@ python3 train.py \
3737
--softmax_variant_attn softmax \
3838
--tensorboard_run_name "$run_name" \
3939
--block_size "$block_size" \
40+
--bias true \
4041
--out_dir "${output_dir}"
4142

4243
python3 train.py \
@@ -55,6 +56,7 @@ python3 train.py \
5556
--softmax_variant_attn relumax \
5657
--tensorboard_run_name "$run_name" \
5758
--block_size "$block_size" \
59+
--bias true \
5860
--out_dir "${output_dir2}"
5961

6062
sleep 3

0 commit comments

Comments
 (0)