diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-03-12 15:39:49 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-03-12 15:39:49 +0800 |
| commit | 2457ac03ca7dee9962d9a8bcb3e0ad0f44ce185a (patch) | |
| tree | 905fa5107d4699da8427536ad07e70fe4de1c52a /models/rgb_part_net.py | |
| parent | 2a7d3c04eab1f3c2e5306d1597399582229a87e5 (diff) | |
| parent | c5b4e984a6830fa2d7abe1183eef02ec00a15433 (diff) | |
Merge branch 'python3.8' into python3.7
Diffstat (limited to 'models/rgb_part_net.py')
| -rw-r--r-- | models/rgb_part_net.py | 2 |
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__() |