summaryrefslogtreecommitdiff
path: root/models/model.py
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2021-02-26 20:17:30 +0800
committerJordan Gong <jordan.gong@protonmail.com>2021-02-26 20:17:30 +0800
commitaf4d28333b8204d6ab74738a8d3b6ae869cb9344 (patch)
treebb2bbc28256cb6290abcf62730a8537544ac9a3c /models/model.py
parent12f4b52df6f4ce8a4c4484146469774e4b158d30 (diff)
parent66d7b0e6f80aa6feb5e4e0af9af7ba78db641f9e (diff)
Merge branch 'python3.8' into python3.7
Diffstat (limited to 'models/model.py')
-rw-r--r--models/model.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/models/model.py b/models/model.py
index d179361..1e22483 100644
--- a/models/model.py
+++ b/models/model.py
@@ -311,7 +311,8 @@ class Model:
)
# Init models
- model_hp = self.hp.get('model', {})
+ model_hp: dict = self.hp.get('model', {}).copy()
+ model_hp.pop('triplet_margins', None)
self.rgb_pn = RGBPartNet(self.in_channels, self.in_size, **model_hp)
# Try to accelerate computation using CUDA or others
self.rgb_pn = self.rgb_pn.to(self.device)