diff options
Diffstat (limited to 'models')
| -rw-r--r-- | models/layers.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/models/layers.py b/models/layers.py index e737df2..a0e35f0 100644 --- a/models/layers.py +++ b/models/layers.py @@ -2,7 +2,6 @@ from typing import Union, Tuple import torch import torch.nn as nn -import torch.nn.functional as F class FocalConv2d(nn.Module): @@ -24,4 +23,4 @@ class FocalConv2d(nn.Module): split_size = h // 2 ** self.halving z = x.split(split_size, dim=2) z = torch.cat([self.conv(_) for _ in z], dim=2) - return F.leaky_relu(z, inplace=True) + return z |