Update README.md

README.md

@@ -4,7 +4,7 @@ This repository contains the implementation of **LWM** (Large Wireless Model), a
 
 ## How to Use
 
-### Step-by-Step Guide
+### LWM Inference
 
 1. **Clone the Repository**
 
@@ -95,7 +95,7 @@ This repository contains the implementation of **LWM** (Large Wireless Model), a
     preprocessed_chs = tokenizer(deepmimo_data, scenario_idxs, gen_raw=True)
     ```
 
-5. **Generate the Dataset for Inference**
+5. **LWM Inference**
 
    Choose the type of data you want to generate from the tokenized dataset, such as `cls_emb`, `channel_emb`, or `raw`:
 
@@ -105,7 +105,9 @@ This repository contains the implementation of **LWM** (Large Wireless Model), a
     dataset = dataset_gen(preprocessed_chs, input_type, model)
     ```
 
-6. **Use the Dataset in Downstream Tasks**
+### Post-processing for Downstream Task
+
+1. **Use the Dataset in Downstream Tasks**
 
    Finally, you can use the generated raw channels and their inferred LWM embeddings in your downstream tasks: