diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-23 22:19:51 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-23 22:19:51 +0800 |
| commit | 507e1d163aaa6ea4be23e7f08ff6ce0ef58c830b (patch) | |
| tree | 1e3c722bb63e3873464296121ec290bd3e64ad14 /models/model.py | |
| parent | 59ccf61fed4d95b7fe91bb9552f0deb2f2c75b76 (diff) | |
Remove the third term in canonical consistency loss
Diffstat (limited to 'models/model.py')
| -rw-r--r-- | models/model.py | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/models/model.py b/models/model.py index cccb6c4..ddb715d 100644 --- a/models/model.py +++ b/models/model.py @@ -54,7 +54,6 @@ class Model: self.total_iters = self.meta.get('total_iters', (80000, 80000, 80000)) self.is_train: bool = True - self.train_size: int = 74 self.in_channels: int = 3 self.pr: Optional[int] = None self.k: Optional[int] = None @@ -147,7 +146,7 @@ class Model: hpm_optim_hp = optim_hp.pop('hpm', {}) fc_optim_hp = optim_hp.pop('fc', {}) sched_hp = self.hp.get('scheduler', {}) - self.rgb_pn = RGBPartNet(self.train_size, self.in_channels, **model_hp) + self.rgb_pn = RGBPartNet(self.in_channels, **model_hp) # Try to accelerate computation using CUDA or others self.rgb_pn = self.rgb_pn.to(self.device) self.optimizer = optim.Adam([ @@ -409,7 +408,6 @@ class Model: self, dataset_config: DatasetConfiguration ) -> Union[CASIAB]: - self.train_size = dataset_config.get('train_size', 74) self.in_channels = dataset_config.get('num_input_channels', 3) self._dataset_sig = self._make_signature( dataset_config, |