Views
No views yet
| Parameter | Value |
|---|---|
| Learning rate | 1e-05 |
| LR scheduler | rex (custom; max_lr=1e-5, min_lr=1e-6, warmup_ratio=0.05) |
| Per-device batch size | 1 |
| Gradient accumulation | 4 |
| Effective batch size | 4 |
| Epochs | 2 |
| Max sequence length | 4096 |
| Optimizer | OptimizerNames.PAGED_ADAMW_8BIT |
| Warmup ratio | 0.05 |
| Max gradient norm | 1.0 |
| Precision | bf16 |
| Loss type | nll |
| Chunked cross-entropy | yes |
| Parameter | Value |
|---|---|
| Rank (r) | 32 |
| Alpha | 8 |
| Target modules | .*language_model.layers.\d+.(self_attn.(q |
| rsLoRA | yes |
| Quantization | 4-bit (nf4) |
| Dataset | Samples | Total tokens | Trainable tokens |
|---|---|---|---|
| brainrot_chatlog.jsonl | 3,232 | 1,115,996 | 1,115,996 |
| erotica_quality_cleaned_20pct_seed42.json | 575 | 2,212,468 | 2,212,468 |
| worm_chapters.json | 658 | 2,268,255 | 2,268,255 |
| counter_signal_training_balanced.jsonl | 1,038 | 4,173,666 | 4,173,666 |
| Total | 5,503 | 9,770,385 | 9,770,385 |
1model_name_or_path: Columbidae/gemma4-31b-pt-embed-it
2data_config: data.yaml
3prepared_dataset: prepared_packed
4output_dir: gemma4-31b-pt-embed-r32a8-textcomp
5attn_implementation: flex_attention
6bf16: true
7gradient_checkpointing: true
8gradient_checkpointing_kwargs:
9 use_reentrant: false
10use_cce: true
11chunked_mlp: true
12chunked_mlp_chunks: 16
13dataloader_num_workers: 2
14dataloader_pin_memory: true
15model_parallel: true
16max_memory:
17 0: 16GiB
18 1: 24GiB
19max_length: 4096
20per_device_train_batch_size: 1
21gradient_accumulation_steps: 4
22pad_to_multiple_of: 128
23use_peft: true
24load_in_4bit: true
25bnb_4bit_quant_type: nf4
26lora_r: 32
27lora_alpha: 8
28lora_dropout: 0.0
29use_rslora: true
30lora_target_modules: .*language_model\.layers\.\d+\.(self_attn\.(q|k|v|o)_proj|mlp\.(gate|up|down)_proj)$
31learning_rate: 1.0e-05
32lr_scheduler_type: rex
33warmup_ratio: 0.05
34weight_decay: 0.0
35max_grad_norm: 1.0
36optim: paged_adamw_8bit
37num_train_epochs: 2
38saves_per_epoch: 2
39save_total_limit: 4
40rolling_save_steps: 30
41rolling_save_total_limit: 1
42logging_steps: 1
43disable_tqdm: false
44report_to: wandb
45run_name: g4-31b-pt-embed-r32a8-textcomp1datasets:
2- path: erotica_quality_cleaned_20pct_seed42.json
3 type: text
4 truncation_strategy: split
5- path: counter_signal_training_balanced.jsonl
6 type: text
7 truncation_strategy: split
8- path: worm_chapters.json
9 type: text
10 truncation_strategy: split
11- path: brainrot_chatlog.jsonl
12 type: text
13 truncation_strategy: split
14shuffle_datasets: true
15shuffle_combined: true
16shuffle_seed: 42
17eval_split: 0
18split_seed: 42
19assistant_only_loss: false