(self, root, test_envs, hparams)
| 235 | CHECKPOINT_FREQ = 1000 |
| 236 | ENVIRONMENTS = ["clip", "info", "paint", "quick", "real", "sketch"] |
| 237 | def __init__(self, root, test_envs, hparams): |
| 238 | self.dir = os.path.join(root, "domain_net/") |
| 239 | super().__init__(self.dir, test_envs, hparams['data_augmentation'], hparams) |
| 240 | |
| 241 | class OfficeHome(MultipleEnvironmentImageFolder): |
| 242 | CHECKPOINT_FREQ = 300 |