Machine-Learning
/
AIIA
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

updated base models MaxPool2D

This commit is contained in:
Falko Victor Habel 2025-01-27 08:39:42 +01:00
parent 29f0d86ff7
commit 3749ba9c5f
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/aiia/model/Model.py
View File

@ -80,8 +80,9 @@ class AIIABaseShared(AIIA):
# Initialize max pooling layer # Initialize max pooling layer
self.max_pool = nn.MaxPool2d( self.max_pool = nn.MaxPool2d(
kernel_size=self.config.kernel_size, kernel_size=1,
padding=1 # Using same padding as in Conv2d layers stride=1,
padding=1
) )
def forward(self, x): def forward(self, x):
@ -117,7 +118,7 @@ class AIIABase(AIIA):
nn.Conv2d(in_channels, self.config.hidden_size, nn.Conv2d(in_channels, self.config.hidden_size,
kernel_size=self.config.kernel_size, padding=1), kernel_size=self.config.kernel_size, padding=1),
getattr(nn, self.config.activation_function)(), getattr(nn, self.config.activation_function)(),
nn.MaxPool2d(kernel_size=2) nn.MaxPool2d(kernel_size=1, stride=1)
]) ])
in_channels = self.config.hidden_size in_channels = self.config.hidden_size