Train and test an AI health assistant in Hausa language
๐ Current Status: Ready
Per device batch size for training. Start with 1 on CPU to avoid OOM.
Number of updates steps to accumulate gradients for. Start with 1 for debugging.
Number of training epochs. Start with 1 for initial tests.
Initial learning rate for the optimizer.
Directory to save the fine-tuned model and tokenizer.