From 1377bc0ea980e5ec3f568def387f3e0a75bcf2a7 Mon Sep 17 00:00:00 2001
From: Paul Best <paul.best@lis-lab.fr>
Date: Wed, 5 Jan 2022 16:25:49 +0100
Subject: [PATCH] fix height pooling for dolphin model

---
 model.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/model.py b/model.py
index 68b2ada..b3ba93d 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
+)
-- 
GitLab