diff --git a/models.py b/models.py index dfba07ad96e133530efd1a0794432e630f074962..abf66e23d20ead2b787fd2b7e7531ac4d89e1a84 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(