From 8217dc798f139532ea710835fbb8d685e493eaf7 Mon Sep 17 00:00:00 2001 From: Siddharth Mishra-Sharma Date: Sat, 16 Mar 2024 15:53:37 -0400 Subject: [PATCH] README update --- README.md | 1 + 1 file changed, 1 insertion(+) diff --git a/README.md b/README.md index aac93a5..dc3b4a1 100644 --- a/README.md +++ b/README.md @@ -34,6 +34,7 @@ mamba env create --file environment.yaml ## Code overview +The code primarily uses Jax. The main components are: - The script for downloading the data is [download_data.py](download_data.py), the summarization script is [summarize.py](summarize.py), and training script is [train.py](train.py). - [notebooks/01_create_dataset.ipynb](notebooks/01_create_dataset.ipynb) is used to create the `tfrecords` data used for training.