github/gpt4all
1
0
Fork 0
mirror of https://github.com/nomic-ai/gpt4all.git synced 2025-07-07 20:41:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: update config

Browse Source
This commit is contained in:
Zach Nussbaum 2023-05-03 21:33:52 +00:00
parent 3fa80f8c09
commit aa6763daa8
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
configs/train/finetune_gptjr.yaml
View File

@ -1,9 +1,10 @@
# model/tokenizer
model_name: "EleutherAI/gpt-j-6B"
tokenizer_name: "EleutherAI/gpt-j-6B"
model_name: "EleutherAI/gpt-j-6b"
tokenizer_name: "EleutherAI/gpt-j-6b"
version: null
gradient_checkpointing: true
save_name: "nomic-ai/gpt-jr-decay-alpha"
push_to_hub: false
encoder_dim: 384
# dataset
@ -11,7 +12,7 @@ streaming: false
num_proc: 64
dataset_path: "/home/paperspace/gpt4all/gpt4all/index/squad_supplemented_train"
max_length: 1024
batch_size: 32
batch_size: 8
pct_test: 0.05
q_column: "question"
a_column: "answers"
@ -23,7 +24,7 @@ lr: 1.0e-4
min_lr: 0
weight_decay: 0.0
eval_every: 50
save_every: 500
save_every: -1
log_grads_every: 100
log_lr_every: 10
output_dir: "ckpts/decay_alpha"
@ -31,6 +32,7 @@ checkpoint: null
lora: false
warmup_steps: 500
num_epochs: 5
debug: false
# logging
wandb: true