(self)
| 1193 | self._init_optimizer() |
| 1194 | |
| 1195 | def _init_optimizer(self): |
| 1196 | self.optimizer = torch.optim.Adam( |
| 1197 | list(self.featurizer.parameters()) + list(self.classifier.parameters()), |
| 1198 | lr=self.hparams["lr"], |
| 1199 | weight_decay=self.hparams["weight_decay"], |
| 1200 | ) |
| 1201 | |
| 1202 | def update(self, minibatches, unlabeled=False): |
| 1203 | assert len(minibatches) == self.num_domains |