Skip to content

Commit 4560846

Browse files
authored
feat(readme): update README.md
1 parent c1e8167 commit 4560846

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

README.md

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -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

2121
1. **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.
3333
3. **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

Comments
 (0)