webcoder-pro / training_config.json
caikybaldo999's picture
Upload WebCoder-100M — step 8000
7070f0a verified
Raw
History Blame Contribute Delete
819 Bytes
{
"model_name": "webcoder-100m",
"output_dir": "/content/webcoder-100m",
"tokenizer_dir": "/content/webcoder-tokenizer",
"seed": 42,
"dataset_id": "bigcode/the-stack-smol-xl",
"languages": [
"html",
"css",
"javascript"
],
"max_samples_per_language": 10000,
"min_chars": 120,
"max_chars": 20000,
"license_filter": true,
"vocab_size": 27664,
"context_length": 1024,
"n_embd": 768,
"n_layer": 11,
"n_head": 12,
"train_batch_size": 8,
"eval_batch_size": 8,
"gradient_accumulation_steps": 4,
"learning_rate": 0.0004,
"weight_decay": 0.1,
"warmup_ratio": 0.03,
"max_steps": 8000,
"time_limit_minutes": 55,
"logging_steps": 20,
"eval_steps": 250,
"save_steps": 500,
"save_total_limit": 3,
"hf_repo_id": "caikybaldo999/webcoder-pro",
"hf_private": false
}