Update README.md

Author: n0obcoder

README.md

@@ -4,27 +4,27 @@ Building a simple Neural Network to solve a Univariate Regression problem in PyT
 The main focus of this repository is to showcase the flow of building and training a basic Neural Network in PyTorch.
 
 
-## Step 1: Generating Synthetic 2D Data
+### Step 1: Generating Synthetic 2D Data
 
 ![Generating Synthetic 2D Data](plots/synthetic_m_and_c.jpg)
 
-## Step 2: Defining the Model Architecture
+### Step 2: Defining the Model Architecture
 
-## Step 3: Defining a Loss Function and an Optimizer
+### Step 3: Defining a Loss Function and an Optimizer
 
-## Step 4: Training the Model and Plotting the Loss
+### Step 4: Training the Model and Plotting the Loss
 
 ![Training the Model and Plotting the Loss](plots/losses.jpg)
 
-## Step 5: Visualizing the Straight Line Learnt by the Model
+### Step 5: Visualizing the Straight Line Learnt by the Model
 
 ![Visualizing the Straight Line Learnt by the Model](plots/learnt_m_and_c.jpg)
 
-## Visualizing the Slope and the Intercept during the Training of the Model
+### Visualizing the Slope and the Intercept during the Training of the Model
 
 ![Visualizing the Slope and the Intercept during the Training of the Model](plots/learning_m_and_c.jpg)
 
-## Important Links
+### Important Links
 
 Medium Blog: https://medium.com/@animesh7pointer/fitting-a-straight-line-on-2d-data-in-pytorch