chlee10 commited on
Commit
68ef990
โ€ข
1 Parent(s): ca8179a

Update README.md

Browse files

# ํ•˜์ดํผ ํŒŒ๋ผ๋ฏธํ„ฐ

# ๋ฐ์ดํ„ฐ์…‹๊ณผ ํ›ˆ๋ จ ํšŸ์ˆ˜์™€ ๊ด€๋ จ๋œ ํ•˜์ดํผ ํŒŒ๋ผ๋ฏธํ„ฐ
batch_size = 16
num_epochs = 1
micro_batch = 1
gradient_accumulation_steps = batch_size // micro_batch

# ํ›ˆ๋ จ ๋ฐฉ๋ฒ•์— ๋Œ€ํ•œ ํ•˜์ดํผ ํŒŒ๋ผ๋ฏธํ„ฐ
cutoff_len = 4096
lr_scheduler = 'cosine'
warmup_ratio = 0.06 # warmup_steps = 100
learning_rate = 4e-4
optimizer = 'adamw_torch'
weight_decay = 0.01
max_grad_norm = 1.0

# LoRA config
lora_r = 16
lora_alpha = 16
lora_dropout = 0.05
lora_target_modules = ["gate_proj", "down_proj", "up_proj"]

# Tokenizer์—์„œ ๋‚˜์˜ค๋Š” input๊ฐ’ ์„ค์ • ์˜ต์…˜
train_on_inputs = False
add_eos_token = False

# NEFTune params
noise_alpha: int = 5

Files changed (1) hide show
  1. README.md +3 -0
README.md CHANGED
@@ -1,3 +1,6 @@
 
 
 
1
  Update @ 2024.03.07
2
 
3
  Open-platypus-SOLAR-10.7B-v1.0
 
1
+ ---
2
+ pipeline_tag: text-generation
3
+ ---
4
  Update @ 2024.03.07
5
 
6
  Open-platypus-SOLAR-10.7B-v1.0