apunkt/doc-to-lora
1
0
Fork 0
mirror of https://github.com/SakanaAI/doc-to-lora.git synced 2026-07-23 17:01:04 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

sakura scripts

Browse source
This commit is contained in:
51616 2025-08-11 15:00:30 +09:00
parent 8bbb5d91a1
commit f209c8b9a1
4 changed files with 23 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

6
scripts/short_ctx/gemma_qa_short_ctx_exp_default.sh
View file

@ -1,7 +1,7 @@
#!/bin/bash
#SBATCH --job-name=ctxlora
#SBATCH --nodes=1
#SBATCH --partition=a3
#SBATCH --partition=sakura-gpu
#SBATCH --gpus=4
#SBATCH --output=slurm_logs/%x-%j.out
#SBATCH --error=slurm_logs/%x-%j.out
@ -25,5 +25,5 @@ default_args=(
# Pass all script arguments to the training command
# $1 comes first, then defaults, then remaining arguments (which can override defaults)
uv run accelerate launch --main_process_port $port \
--num_processes=4 --gpu_ids all train.py $1 "${default_args[@]}" "${@:2}"
uv run accelerate launch --config_file accelerate_config.yaml --main_process_port $port \
--num_processes=4 --gpu_ids all train.py $1 "${default_args[@]}" "${@:2}"

6
scripts/short_ctx/gemma_qa_short_ctx_exp_default_8_gpus.sh
View file

@ -1,7 +1,7 @@
#!/bin/bash
#SBATCH --job-name=ctxlora
#SBATCH --nodes=1
#SBATCH --partition=a3
#SBATCH --partition=sakura-gpu
#SBATCH --gpus=8
#SBATCH --output=slurm_logs/%x-%j.out
#SBATCH --error=slurm_logs/%x-%j.out
@ -25,5 +25,5 @@ default_args=(
# Pass all script arguments to the training command
# $1 comes first, then defaults, then remaining arguments (which can override defaults)
uv run accelerate launch --main_process_port $port \
--num_processes=8 --gpu_ids all train.py $1 "${default_args[@]}" "${@:2}"
uv run accelerate launch --config_file accelerate_config.yaml --main_process_port $port \
--num_processes=8 --gpu_ids all train.py $1 "${default_args[@]}" "${@:2}"