diff options
Diffstat (limited to 'models/hpm.py')
| -rw-r--r-- | models/hpm.py | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/models/hpm.py b/models/hpm.py index f387154..1773f56 100644 --- a/models/hpm.py +++ b/models/hpm.py @@ -1,5 +1,3 @@ -from typing import Tuple - import torch import torch.nn as nn from torchvision.models import resnet50 @@ -10,7 +8,7 @@ from models.layers import HorizontalPyramidPooling class HorizontalPyramidMatching(nn.Module): def __init__( self, - scales: Tuple[int] = (1, 2, 4, 8), + scales: tuple[int, ...] = (1, 2, 4, 8), out_channels: int = 256, use_avg_pool: bool = False, **kwargs |