gemma-4-e2b-uiux-lora / training_args.json
RayyTien's picture
Upload Gemma 4 E2B UI/UX LoRA adapter
0adde74 verified
Raw
History Blame Contribute Delete
577 Bytes
{
"base_model": "mlx-community/gemma-4-e2b-it-4bit",
"local_training_model": "./models/gemma-4-e2b-it-4bit",
"adapter_path": "adapters/gemma-e2b-uiux-v2",
"dataset": "./data-uiux",
"framework": "mlx-vlm",
"train_command": "MODEL=./models/gemma-4-e2b-it-4bit DATASET=./data-uiux ITERS=600 ADAPTER_PATH=./adapters/gemma-e2b-uiux-v2 STEPS_PER_SAVE=1000 ./scripts/train_smoke.sh",
"iterations": 600,
"max_seq_length": 1024,
"batch_size": 1,
"lora_rank": 8,
"lora_alpha": 16,
"learning_rate": 0.00002,
"peak_memory_gb": 8.984,
"trained_tokens": 119495
}