diff options
author | Jordan Gong <jordan.gong@protonmail.com> | 2021-02-15 12:06:45 +0800 |
---|---|---|
committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-02-15 12:09:14 +0800 |
commit | 6a4f1cdb88d1853beb410b9da29e43069688dba7 (patch) | |
tree | 6ff66d9cfa290b3f8831580f46a526dc8325a691 /models/hpm.py | |
parent | cf59a1f8339e937e31281907a61f484af99ce955 (diff) | |
parent | 24b5968bfc5799e44c9bbbc00e3a9be00f4509ac (diff) |
Merge branch 'python3.8' into data_parallel_py3.8
# Conflicts:
# models/model.py
# models/rgb_part_net.py
Diffstat (limited to 'models/hpm.py')
-rw-r--r-- | models/hpm.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/models/hpm.py b/models/hpm.py index 9879cfb..b49be3a 100644 --- a/models/hpm.py +++ b/models/hpm.py @@ -1,3 +1,5 @@ +from typing import Tuple + import torch import torch.nn as nn @@ -10,7 +12,7 @@ class HorizontalPyramidMatching(nn.Module): in_channels: int, out_channels: int = 128, use_1x1conv: bool = False, - scales: tuple[int, ...] = (1, 2, 4), + scales: Tuple[int, ...] = (1, 2, 4), use_avg_pool: bool = True, use_max_pool: bool = False, **kwargs |