From dbd3050a71cd34f0b163dcb4d828d3071f4bede2 Mon Sep 17 00:00:00 2001
From: Danish Syed <syeddanish.ece@gmail.com>
Date: Thu, 6 Jun 2024 01:10:50 -0500
Subject: [PATCH] Update README.md

---
 README.md | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/README.md b/README.md
index 67996da..e489bf3 100644
--- a/README.md
+++ b/README.md
@@ -61,6 +61,10 @@ To customize various aspects of the training process, modify the following confi
 2. **Model Customization:** Modify `configs/model/plant_traits_model.yaml` to adjust model configurations.
 3. **Experiment Customization:** Modify `configs/experiment/plant_traits.yaml` to change overall experiment settings.
 
+### Modifying Source Code
+1. **Dataset Modification:** Modify `src/fgvc/data/plant_traits_data.py`
+2. **Model Customization:** Modify `src/fgvc/models/plant_traits_model.py`
+
 ## Acknowledgement
 This experimentation template is generated from [ashleve/lightning-hydra-template](https://github.com/ashleve/lightning-hydra-template). For more details about the template, please refer to their documentation.