release code
This commit is contained in:
@@ -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
|
||||
@@ -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"
|
||||
@@ -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"
|
||||
12
Dassl.ProGrad.pytorch/configs/trainers/dg/vanilla/pacs.yaml
Normal file
12
Dassl.ProGrad.pytorch/configs/trainers/dg/vanilla/pacs.yaml
Normal file
@@ -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"
|
||||
Reference in New Issue
Block a user