From edfc6790e987f06ddc83b29b85518fbbbcb9fe9a Mon Sep 17 00:00:00 2001
From: HantingChen <40243544+HantingChen@users.noreply.github.com>
Date: Wed, 31 May 2023 09:50:11 +0800
Subject: [PATCH] Update README.md

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

diff --git a/README.md b/README.md
index 10954b7..dbbfd25 100644
--- a/README.md
+++ b/README.md
@@ -75,6 +75,36 @@ pip install tensorboardX
 pip install terminaltables
 ```
 
+## Dataset Preparation
+
+Download the [ImageNet-1K](http://image-net.org/) classification dataset and structure the data as follows:
+```
+/path/to/imagenet-1k/
+  train/
+    class1/
+      img1.jpeg
+    class2/
+      img2.jpeg
+  val/
+    class1/
+      img3.jpeg
+    class2/
+      img4.jpeg
+```
+
+For pre-training on [ImageNet-22K](http://image-net.org/), download the dataset and structure the data as follows:
+```
+/path/to/imagenet-22k/
+  class1/
+    img1.jpeg
+  class2/
+    img2.jpeg
+  class3/
+    img3.jpeg
+  class4/
+    img4.jpeg
+```
+
 ## Testing
 
 We give an example evaluation command  for VanillaNet-5: