Skip to content

Commit 3923edd

Browse files
Update README.md
1 parent 8fdbe9d commit 3923edd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ Reddit User Mimic Bot is an end-to-end Python project that fine‑tunes a BART m
44

55
## Features
66
- **LoRA Training** – fine-tune `facebook/bart-base` using Low-Rank Adaptation on scraped Reddit Q&A data.
7-
- **Data Pipeline** – scripts to scrape Reddit posts with filters for gathering better quality data, preprocess them, split into train/validation/test sets, tokenize and reformat.
7+
- **Data Pipeline** – scripts to scrape Reddit posts with filters for gathering better quality data, preprocess them, split into train/validation/test sets, tokenize and reformat; leverages data collator for dynamic batching and tensor‑core‑friendly padding.
88
- **Hugging Face Integration** – optional model upload to the Hugging Face Hub.
99
- **Logging and Experiment Tracking** – train/loss, val/loss tracked via Weights & Biases.
1010
- **Model Efficiency** – utilize early stopping and Scaled Dot-Product Attention (SDPA) to optimize training efficiency and manage compute resources effectively.

0 commit comments

Comments
 (0)