DOC: update README.md

msafari18 · msafari18 · commit e5bc6b722777 · 2025-04-11T16:08:48.000-04:00
diff --git a/CC_requirements.txt b/CC_requirements.txt
@@ -14,8 +14,8 @@ scikit-learn==1.3+computecanada
 scipy==1.12+computecanada
 seaborn
 torch==2.1.1+computecanada
+torchdata==0.7.0+computecanada
 torchtext==0.16.1+computecanada
 transformers==4.29.2+computecanada
 umap-learn==0.5.6+computecanada
-torchdata==0.7.0+computecanada
 wandb
diff --git a/README.md b/README.md
@@ -12,6 +12,7 @@ A PyTorch implementation of BarcodeMAE, a model for enhancing DNA foundation mod
 
 ### Setup
 
+
 0. Clone this repository 
 1. Install the required libraries
 
@@ -32,6 +33,7 @@ python data_split.py BIOSCAN-5M_Dataset_metadata.tsv
 
 ### Reproducing the results
 
+
 1. Download the checkpoint and copy it to the model_checkpoints directory
 2. Run KNN evaluation
 
@@ -46,6 +48,7 @@ python barcodebert/knn_probing.py \
 
 ### Pretraining from scratch
 
+
 1. Run pretraining 
 
 ```shell
diff --git a/requirements.txt b/requirements.txt
@@ -14,8 +14,8 @@ scikit-learn==1.3
 scipy==1.12
 seaborn
 torch==2.1.1
+torchdata==0.7.0
 torchtext==0.16.1
 transformers==4.29.2
 umap-learn==0.5.6
-torchdata==0.7.0
 wandb
