From eb9606adb0e51193caccadbbc091da0dd59af0c3 Mon Sep 17 00:00:00 2001
From: Eternal Reclaimer <98760976+kyegomez@users.noreply.github.com>
Date: Wed, 27 Dec 2023 17:50:27 -0500
Subject: [PATCH] Update stochasticskipblock.md

---
 docs/zeta/nn/modules/stochasticskipblock.md | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/docs/zeta/nn/modules/stochasticskipblock.md b/docs/zeta/nn/modules/stochasticskipblock.md
index f6c7a72d..a7ef7941 100644
--- a/docs/zeta/nn/modules/stochasticskipblock.md
+++ b/docs/zeta/nn/modules/stochasticskipblock.md
@@ -63,6 +63,7 @@ First, you need to import the necessary module:
 import torch
 import torch.nn as nn
 from torch.nn.functional import relu
+from zeta.nn import StochasticSkipBlock
 ```
 
 Now, you need to define the architecture of the model:
@@ -125,6 +126,8 @@ This shows how to train the model using StochasticSkipBlock module. Please note,
 from torch.optim import SGD
 from torch.nn.functional import binary_cross_entropy
 import torch.optim as optim
+from zeta.nn import StochasticSkipBlock
+
 
 #initiate model
 model = MyModel()