mirror of
https://github.com/SakanaAI/doc-to-lora.git
synced 2026-07-23 17:01:04 +02:00
scripts
This commit is contained in:
parent
29719f4d19
commit
2b824ed813
3 changed files with 6 additions and 6 deletions
|
|
@ -13,9 +13,9 @@ echo "Using port: $port"
|
|||
uv run accelerate launch --main_process_port $port \
|
||||
--num_processes=4 --gpu_ids all intx_sft.py $1 \
|
||||
--model_name_or_path=google/gemma-2-2b-it \
|
||||
--max_steps=10_000 \
|
||||
--max_steps=30_000 \
|
||||
--per_device_train_batch_size=-1 \
|
||||
--gradient_accumulation_steps=2 \
|
||||
--gradient_accumulation_steps=4 \
|
||||
--per_device_eval_batch_size=64 \
|
||||
--target_modules=down_proj \
|
||||
--num_blocks=8 \
|
||||
|
|
|
|||
|
|
@ -13,9 +13,9 @@ echo "Using port: $port"
|
|||
uv run accelerate launch --main_process_port $port \
|
||||
--num_processes=4 --gpu_ids all intx_sft.py $1 \
|
||||
--model_name_or_path=google/gemma-2-2b-it \
|
||||
--max_steps=10_000 \
|
||||
--max_steps=30_000 \
|
||||
--per_device_train_batch_size=-1 \
|
||||
--gradient_accumulation_steps=4 \
|
||||
--gradient_accumulation_steps=8 \
|
||||
--per_device_eval_batch_size=32 \
|
||||
--target_modules=down_proj \
|
||||
--num_blocks=8 \
|
||||
|
|
|
|||
|
|
@ -13,9 +13,9 @@ echo "Using port: $port"
|
|||
uv run accelerate launch --main_process_port $port \
|
||||
--num_processes=4 --gpu_ids all intx_sft.py $1 \
|
||||
--model_name_or_path=google/gemma-2-2b-it \
|
||||
--max_steps=10_000 \
|
||||
--max_steps=30_000 \
|
||||
--per_device_train_batch_size=-1 \
|
||||
--gradient_accumulation_steps=4 \
|
||||
--gradient_accumulation_steps=8 \
|
||||
--per_device_eval_batch_size=32 \
|
||||
--target_modules=down_proj \
|
||||
--num_blocks=8 \
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue