release code

Dassl.ProGrad.pytorch/configs/trainers/dg/dael/digits_dg.yaml

@@ -0,0 +1,16 @@
+DATALOADER:
+  TRAIN_X:
+    SAMPLER: "RandomDomainSampler"
+    BATCH_SIZE: 120
+  TEST:
+    BATCH_SIZE: 100
+
+OPTIM:
+  NAME: "sgd"
+  LR: 0.05
+  STEPSIZE: [20]
+  MAX_EPOCH: 50
+
+TRAINER:
+  DAEL:
+    STRONG_TRANSFORMS: ["randaugment2", "normalize"]

Dassl.ProGrad.pytorch/configs/trainers/dg/dael/office_home_dg.yaml

@@ -0,0 +1,16 @@
+DATALOADER:
+  TRAIN_X:
+    SAMPLER: "RandomDomainSampler"
+    BATCH_SIZE: 30
+  TEST:
+    BATCH_SIZE: 100
+
+OPTIM:
+  NAME: "sgd"
+  LR: 0.002
+  MAX_EPOCH: 40
+  LR_SCHEDULER: "cosine"
+
+TRAINER:
+  DAEL:
+    STRONG_TRANSFORMS: ["random_flip", "cutout", "randaugment2", "normalize"]

Dassl.ProGrad.pytorch/configs/trainers/dg/dael/pacs.yaml

@@ -0,0 +1,16 @@
+DATALOADER:
+  TRAIN_X:
+    SAMPLER: "RandomDomainSampler"
+    BATCH_SIZE: 30
+  TEST:
+    BATCH_SIZE: 100
+
+OPTIM:
+  NAME: "sgd"
+  LR: 0.002
+  MAX_EPOCH: 40
+  LR_SCHEDULER: "cosine"
+
+TRAINER:
+  DAEL:
+    STRONG_TRANSFORMS: ["random_flip", "cutout", "randaugment2", "normalize"]

Dassl.ProGrad.pytorch/configs/trainers/dg/ddaig/digits_dg.yaml

@@ -0,0 +1,20 @@
+INPUT:
+  PIXEL_MEAN: [0., 0., 0.]
+  PIXEL_STD: [1., 1., 1.]
+
+DATALOADER:
+  TRAIN_X:
+    BATCH_SIZE: 128
+  TEST:
+    BATCH_SIZE: 128
+
+OPTIM:
+  NAME: "sgd"
+  LR: 0.05
+  STEPSIZE: [20]
+  MAX_EPOCH: 50
+
+TRAINER:
+  DDAIG:
+    G_ARCH: "fcn_3x32_gctx"
+    LMDA: 0.3

Dassl.ProGrad.pytorch/configs/trainers/dg/ddaig/office_home_dg.yaml

@@ -0,0 +1,21 @@
+INPUT:
+  PIXEL_MEAN: [0., 0., 0.]
+  PIXEL_STD: [1., 1., 1.]
+
+DATALOADER:
+  TRAIN_X:
+    BATCH_SIZE: 16
+  TEST:
+    BATCH_SIZE: 16
+
+OPTIM:
+  NAME: "sgd"
+  LR: 0.0005
+  STEPSIZE: [20]
+  MAX_EPOCH: 25
+
+TRAINER:
+  DDAIG:
+    G_ARCH: "fcn_3x64_gctx"
+    WARMUP: 3
+    LMDA: 0.3

Dassl.ProGrad.pytorch/configs/trainers/dg/ddaig/pacs.yaml

@@ -0,0 +1,21 @@
+INPUT:
+  PIXEL_MEAN: [0., 0., 0.]
+  PIXEL_STD: [1., 1., 1.]
+
+DATALOADER:
+  TRAIN_X:
+    BATCH_SIZE: 16
+  TEST:
+    BATCH_SIZE: 16
+
+OPTIM:
+  NAME: "sgd"
+  LR: 0.0005
+  STEPSIZE: [20]
+  MAX_EPOCH: 25
+
+TRAINER:
+  DDAIG:
+    G_ARCH: "fcn_3x64_gctx"
+    WARMUP: 3
+    LMDA: 0.3

Dassl.ProGrad.pytorch/configs/trainers/dg/vanilla/digits_dg.yaml

@@ -0,0 +1,15 @@
+DATALOADER:
+  TRAIN_X:
+    BATCH_SIZE: 128
+  TEST:
+    BATCH_SIZE: 100
+  NUM_WORKERS: 8
+
+OPTIM:
+  NAME: "sgd"
+  LR: 0.05
+  STEPSIZE: [20]
+  MAX_EPOCH: 50
+
+TRAIN:
+  PRINT_FREQ: 20

Dassl.ProGrad.pytorch/configs/trainers/dg/vanilla/mini_domainnet.yaml

@@ -0,0 +1,12 @@
+DATALOADER:
+  NUM_WORKERS: 8
+  TRAIN_X:
+    BATCH_SIZE: 128
+  TEST:
+    BATCH_SIZE: 128
+
+OPTIM:
+  NAME: "sgd"
+  LR: 0.005
+  MAX_EPOCH: 60
+  LR_SCHEDULER: "cosine"

Dassl.ProGrad.pytorch/configs/trainers/dg/vanilla/office_home_dg.yaml

@@ -0,0 +1,12 @@
+DATALOADER:
+  TRAIN_X:
+    BATCH_SIZE: 64
+  TEST:
+    BATCH_SIZE: 100
+  NUM_WORKERS: 8
+
+OPTIM:
+  NAME: "sgd"
+  LR: 0.001
+  MAX_EPOCH: 50
+  LR_SCHEDULER: "cosine"

Dassl.ProGrad.pytorch/configs/trainers/dg/vanilla/pacs.yaml

@@ -0,0 +1,12 @@
+DATALOADER:
+  TRAIN_X:
+    BATCH_SIZE: 64
+  TEST:
+    BATCH_SIZE: 100
+  NUM_WORKERS: 8
+
+OPTIM:
+  NAME: "sgd"
+  LR: 0.001
+  MAX_EPOCH: 50
+  LR_SCHEDULER: "cosine"