Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
FinGPT-glm_pytorch
Commits
2df98a70
Commit
2df98a70
authored
May 10, 2024
by
wanglch
Browse files
Delete single_dcu_train.sh
parent
90079609
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
0 additions
and
32 deletions
+0
-32
single_dcu_train.sh
single_dcu_train.sh
+0
-32
No files found.
single_dcu_train.sh
deleted
100644 → 0
View file @
90079609
#!/bin/bash
CUDA_VISIBLE_DEVICES
=
0 python /LLaMA-Factory-main/src/train_bash.py
\
--stage
sft
\
--do_train
\
--model_name_or_path
THUDM/chatglm2-6b
\
--dataset
fingpt_sentiment
\
--dataset_dir
/LLaMA-Factory-main/data
\
--template
default
\
--finetuning_type
lora
\
--lora_target
all
\
--output_dir
/saves/FinGPT/lora/sft
\
--overwrite_cache
\
--overwrite_output_dir
\
--cutoff_len
1024
\
--preprocessing_num_workers
16
\
--per_device_train_batch_size
1
\
--per_device_eval_batch_size
1
\
--gradient_accumulation_steps
8
\
--lr_scheduler_type
cosine
\
--logging_steps
10
\
--warmup_steps
20
\
--save_steps
100
\
--eval_steps
100
\
--evaluation_strategy
steps
\
--load_best_model_at_end
\
--learning_rate
5e-5
\
--num_train_epochs
3.0
\
--max_samples
3000
\
--val_size
0.1
\
--plot_loss
\
--fp16
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment