@@ -12,14 +12,14 @@ Flare AI Kit template for Social AI Agents.
1212 | --------------------- | -------------------------------------------------------------------------- | ------------------------------------ |
1313 | ` tuned_model_name ` | Name of the newly tuned model. | ` pugo-hilion ` |
1414 | ` tuning_source_model ` | Name of the foundational model to tune on. | ` models/gemini-1.5-flash-001-tuning ` |
15- | ` epoch_count ` | Number of tuning epochs to run. An epoch is a pass over the whole dataset. | ` 100 ` |
15+ | ` epoch_count ` | Number of tuning epochs to run. An epoch is a pass over the whole dataset. | ` 30 ` |
1616 | ` batch_size ` | Number of examples to use in each training batch. | ` 4 ` |
1717 | ` learning_rate ` | Step size multiplier for the gradient updates. | ` 0.001 ` |
1818
1919### Fine tuning a model over a dataset
2020
21211 . ** Prepare a dataset:**
22- An example dataset is provided in ` src/data/training_data.json ` , which consists of tweet from
22+ An example dataset is provided in ` src/data/training_data.json ` , which consists of tweets from
2323 [ Hugo Philion's X] ( https://x.com/HugoPhilion ) account. You can use any publicly available dataset
2424 for model fine-tuning.
2525
@@ -33,8 +33,10 @@ Flare AI Kit template for Social AI Agents.
33333 . ** Observe loss parameters:**
3434 After tuning in complete, a training loss PNG will be saved in the root folder corresponding to the new model.
3535 Ideally the loss should minimize to near 0 after several training epochs.
36+
37+ ![ pugo-hilion_mean_loss] ( https://github.com/user-attachments/assets/f6c4d82b-678a-4ae5-bfb7-39dc59e1103d )
3638
37- 4 . ** Test the new model**
39+ 5 . ** Test the new model**
3840 Select the new tuned model and compare it against a set of prompting techniques (zero-shot, few-shot and chain-of-thought):
3941
4042 ``` bash
0 commit comments