Skip to content

Commit

Permalink
add
Browse files Browse the repository at this point in the history
  • Loading branch information
sungho.park committed Mar 16, 2023
1 parent d10b578 commit b910a95
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions train.sh
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
python train.py \
--model_name_or_path decapoda-research/llama-7b-hf \
--model_name_or_path $model_path \
--dataset_name alpaca_data.hf \
--is_dataset_from_disk True \
--per_device_train_batch_size 8 \
--per_device_eval_batch_size 8 \
--do_train \
--do_eval \
--output_dir test-clm
--output_dir $output_dir

0 comments on commit b910a95

Please sign in to comment.