summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2021-02-19 10:17:43 +0800
committerJordan Gong <jordan.gong@protonmail.com>2021-02-19 10:17:43 +0800
commit5b3a324df1ee4c3789667f0f98bdfa5187393680 (patch)
treeedd304cc45304947e9e20042e95335e2af350f12
parent8012fce3e595aad31f4a52dc316b46e558234dff (diff)
parent6b3b281bb91f9550969322e027c15f348750c726 (diff)
Merge branch 'python3.8' into data_parallel_py3.8
-rw-r--r--models/auto_encoder.py6
-rw-r--r--models/model.py2
-rw-r--r--models/rgb_part_net.py2
-rw-r--r--requirements.txt2
4 files changed, 6 insertions, 6 deletions
diff --git a/models/auto_encoder.py b/models/auto_encoder.py
index c6bc52f..e17caed 100644
--- a/models/auto_encoder.py
+++ b/models/auto_encoder.py
@@ -13,7 +13,7 @@ class Encoder(nn.Module):
def __init__(
self,
in_channels: int = 3,
- frame_size: tuple[int, int] = (64, 48),
+ frame_size: Tuple[int, int] = (64, 48),
feature_channels: int = 64,
output_dims: Tuple[int, int, int] = (128, 128, 64)
):
@@ -78,7 +78,7 @@ class Decoder(nn.Module):
self,
input_dims: Tuple[int, int, int] = (128, 128, 64),
feature_channels: int = 64,
- feature_size: tuple[int, int] = (4, 3),
+ feature_size: Tuple[int, int] = (4, 3),
out_channels: int = 3,
):
super().__init__()
@@ -127,7 +127,7 @@ class AutoEncoder(nn.Module):
def __init__(
self,
channels: int = 3,
- frame_size: tuple[int, int] = (64, 48),
+ frame_size: Tuple[int, int] = (64, 48),
feature_channels: int = 64,
embedding_dims: Tuple[int, int, int] = (128, 128, 64)
):
diff --git a/models/model.py b/models/model.py
index 3d619fe..75f243c 100644
--- a/models/model.py
+++ b/models/model.py
@@ -55,7 +55,7 @@ class Model:
self.is_train: bool = True
self.in_channels: int = 3
- self.in_size: tuple[int, int] = (64, 48)
+ self.in_size: Tuple[int, int] = (64, 48)
self.pr: Optional[int] = None
self.k: Optional[int] = None
diff --git a/models/rgb_part_net.py b/models/rgb_part_net.py
index 80b3e17..538ca53 100644
--- a/models/rgb_part_net.py
+++ b/models/rgb_part_net.py
@@ -13,7 +13,7 @@ class RGBPartNet(nn.Module):
def __init__(
self,
ae_in_channels: int = 3,
- ae_in_size: tuple[int, int] = (64, 48),
+ ae_in_size: Tuple[int, int] = (64, 48),
ae_feature_channels: int = 64,
f_a_c_p_dims: Tuple[int, int, int] = (128, 128, 64),
hpm_use_1x1conv: bool = False,
diff --git a/requirements.txt b/requirements.txt
index f64085a..4d30e17 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,6 @@
torch~=1.7.1
torchvision~=0.8.0a0+ecf4e9c
numpy~=1.19.4
-tqdm~=4.56.0
+tqdm~=4.57.0
Pillow~=8.1.0
scikit-learn~=0.24.0 \ No newline at end of file