From 87ba81e0c4bcd230bc8664e6e69a5be0b18046ea Mon Sep 17 00:00:00 2001
From: lamipaul <paulobest25@gmail.com>
Date: Mon, 1 Aug 2022 20:30:09 +0200
Subject: [PATCH] minor fix megptera maxPool
---
models.py | 3 +--
1 file changed, 1 insertion(+), 2 deletions(-)
diff --git a/models.py b/models.py
index dfba07a..abf66e2 100644
--- a/models.py
+++ b/models.py
@@ -37,8 +37,7 @@ get = {
nn.BatchNorm2d(64),
nn.LeakyReLU(0.01),
nn.Dropout(p=.5),
- nn.Conv2d(64, 1, 1, bias=False),
- nn.MaxPool2d((6, 1))
+ nn.Conv2d(64, 1, 1, bias=False)
)
),
'delphinid' : nn.Sequential(
--
GitLab