From 79706a2cbfb1e725930a78c6448eda5af6de0aed Mon Sep 17 00:00:00 2001
From: Ahmed Besbes <ahmed.besbes@hotmail.com>
Date: Tue, 22 Oct 2019 00:06:56 +0200
Subject: [PATCH] adding dropout to input

---
 src/model.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/model.py b/src/model.py
index aabf559..355bc4c 100644
--- a/src/model.py
+++ b/src/model.py
@@ -90,6 +90,7 @@ def _get_conv_output(self, shape):
     # forward
 
     def forward(self, x):
+        x = nn.Dropout2d(0.1)(x)
         x = x.transpose(1, 2)
         x = self.conv1(x)
         x = self.conv2(x)
@@ -102,3 +103,4 @@ def forward(self, x):
         x = self.fc2(x)
         x = self.fc3(x)
         return x
+