Skip to content
Snippets Groups Projects
Commit a82fcf16 authored by zqwerty's avatar zqwerty
Browse files

update script

parent 109d507d
No related branches found
No related tags found
No related merge requests found
...@@ -24,7 +24,8 @@ num_train_epochs=10 ...@@ -24,7 +24,8 @@ num_train_epochs=10
python ../create_data.py --tasks ${task_name} --datasets ${dataset_name} --speaker ${speaker} --context_window_size ${context_window_size} python ../create_data.py --tasks ${task_name} --datasets ${dataset_name} --speaker ${speaker} --context_window_size ${context_window_size}
python ../run_seq2seq.py \ python -m torch.distributed.launch \
--nproc_per_node ${n_gpus} ../run_seq2seq.py \
--task_name ${task_name} \ --task_name ${task_name} \
--train_file ${train_file} \ --train_file ${train_file} \
--source_column ${source_column} \ --source_column ${source_column} \
...@@ -48,7 +49,8 @@ python ../run_seq2seq.py \ ...@@ -48,7 +49,8 @@ python ../run_seq2seq.py \
--adafactor \ --adafactor \
--gradient_checkpointing --gradient_checkpointing
python ../run_seq2seq.py \ python -m torch.distributed.launch \
--nproc_per_node ${n_gpus} ../run_seq2seq.py \
--task_name ${task_name} \ --task_name ${task_name} \
--test_file ${test_file} \ --test_file ${test_file} \
--source_column ${source_column} \ --source_column ${source_column} \
......
...@@ -24,7 +24,8 @@ num_train_epochs=10 ...@@ -24,7 +24,8 @@ num_train_epochs=10
python ../create_data.py --tasks ${task_name} --datasets ${dataset_name} --speaker ${speaker} --context_window_size ${context_window_size} python ../create_data.py --tasks ${task_name} --datasets ${dataset_name} --speaker ${speaker} --context_window_size ${context_window_size}
python ../run_seq2seq.py \ python -m torch.distributed.launch \
--nproc_per_node ${n_gpus} ../run_seq2seq.py \
--task_name ${task_name} \ --task_name ${task_name} \
--train_file ${train_file} \ --train_file ${train_file} \
--source_column ${source_column} \ --source_column ${source_column} \
...@@ -48,7 +49,8 @@ python ../run_seq2seq.py \ ...@@ -48,7 +49,8 @@ python ../run_seq2seq.py \
--adafactor \ --adafactor \
--gradient_checkpointing --gradient_checkpointing
python ../run_seq2seq.py \ python -m torch.distributed.launch \
--nproc_per_node ${n_gpus} ../run_seq2seq.py \
--task_name ${task_name} \ --task_name ${task_name} \
--test_file ${test_file} \ --test_file ${test_file} \
--source_column ${source_column} \ --source_column ${source_column} \
......
...@@ -24,7 +24,8 @@ num_train_epochs=10 ...@@ -24,7 +24,8 @@ num_train_epochs=10
python ../create_data.py --tasks ${task_name} --datasets ${dataset_name} --speaker ${speaker} --context_window_size ${context_window_size} python ../create_data.py --tasks ${task_name} --datasets ${dataset_name} --speaker ${speaker} --context_window_size ${context_window_size}
python ../run_seq2seq.py \ python -m torch.distributed.launch \
--nproc_per_node ${n_gpus} ../run_seq2seq.py \
--task_name ${task_name} \ --task_name ${task_name} \
--train_file ${train_file} \ --train_file ${train_file} \
--source_column ${source_column} \ --source_column ${source_column} \
...@@ -48,7 +49,8 @@ python ../run_seq2seq.py \ ...@@ -48,7 +49,8 @@ python ../run_seq2seq.py \
--adafactor \ --adafactor \
--gradient_checkpointing --gradient_checkpointing
python ../run_seq2seq.py \ python -m torch.distributed.launch \
--nproc_per_node ${n_gpus} ../run_seq2seq.py \
--task_name ${task_name} \ --task_name ${task_name} \
--test_file ${test_file} \ --test_file ${test_file} \
--source_column ${source_column} \ --source_column ${source_column} \
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment