summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--models/rgb_part_net.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/models/rgb_part_net.py b/models/rgb_part_net.py
index 3a5777e..6e72e38 100644
--- a/models/rgb_part_net.py
+++ b/models/rgb_part_net.py
@@ -20,7 +20,7 @@ class RGBPartNet(nn.Module):
hpm_use_max_pool: bool = True,
tfa_squeeze_ratio: int = 4,
tfa_num_parts: int = 16,
- embedding_dims: tuple[int] = (256, 256),
+ embedding_dims: Tuple[int] = (256, 256),
image_log_on: bool = False
):
super().__init__()