diff --git a/model.py b/model.py
index 68b2ada10bbbea1e02871e037a89d0940f42e765..b3ba93dcf74976543e967fcff696abe5d7e0b097 100644
--- a/model.py
+++ b/model.py
@@ -225,5 +225,6 @@ delphi_model = nn.Sequential(nn.Sequential(
   nn.LeakyReLU(0.01),
   nn.Dropout(p=.5),
   nn.Conv2d(64, 1, 1, bias=False),
+  nn.MaxPool2d((6, 1))
   )
-)
\ No newline at end of file
+)