wip deepseek train

This commit is contained in:
Niels Geens
2025-03-03 16:57:04 +01:00
parent b64f8d7d33
commit 3ea3239a9b
21 changed files with 175 additions and 7058 deletions

View File

@@ -4,6 +4,11 @@ model:
params:
learning_rate: 1e-5
epochs: 3
max_seq_length: 1024
quantization: "4bit" # Options: "none", "4bit", "8bit"
per_device_batch_size: 1
gradient_accumulation_steps: 8
mixed_precision: "no" # Options: "no", "fp16", "bf16"
data:
- "/root/sia/training/clean_start/"
- "/root/sia/training/delete_indicated_entries/"