This commit is contained in:
51616 2025-07-29 07:00:03 +00:00
parent 29719f4d19
commit 2b824ed813
3 changed files with 6 additions and 6 deletions

View file

@ -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 \

View file

@ -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 \

View file

@ -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 \