diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-21 23:47:11 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-21 23:47:11 +0800 |
| commit | d750dd9dafe3cda3b1331ad2bfecb53c8c2b1267 (patch) | |
| tree | 1f4c5eeae4f3626295d4968b1a2d998c524a7d7f /models/model.py | |
| parent | 0b76205ecef02dd62ef2fbc8e12d9389b7cf7868 (diff) | |
A type hint fix
Diffstat (limited to 'models/model.py')
| -rw-r--r-- | models/model.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/models/model.py b/models/model.py index bed28a5..f4604c8 100644 --- a/models/model.py +++ b/models/model.py @@ -133,7 +133,7 @@ class Model: dataloader = self._parse_dataloader_config(dataset, dataloader_config) # Prepare for model, optimizer and scheduler model_hp = self.hp.get('model', {}) - optim_hp: dict = self.hp.get('optimizer', {}).copy() + optim_hp: Dict = self.hp.get('optimizer', {}).copy() ae_optim_hp = optim_hp.pop('auto_encoder', {}) pn_optim_hp = optim_hp.pop('part_net', {}) hpm_optim_hp = optim_hp.pop('hpm', {}) |