(self, input_shape, num_classes, num_domains, hparams)
| 936 | """ |
| 937 | |
| 938 | def __init__(self, input_shape, num_classes, num_domains, hparams): |
| 939 | super(ANDMask, self).__init__(input_shape, num_classes, num_domains, hparams) |
| 940 | |
| 941 | self.tau = hparams["tau"] |
| 942 | |
| 943 | def update(self, minibatches, unlabeled=None): |
| 944 | mean_loss = 0 |