(self, input_shape, num_classes, num_domains, hparams)
| 910 | """ |
| 911 | |
| 912 | def __init__(self, input_shape, num_classes, num_domains, hparams): |
| 913 | super(SD, self).__init__(input_shape, num_classes, num_domains, hparams) |
| 914 | self.sd_reg = hparams["sd_reg"] |
| 915 | |
| 916 | def update(self, minibatches, unlabeled=None): |
| 917 | all_x = torch.cat([x for x, y in minibatches]) |
no outgoing calls
no test coverage detected