release code
This commit is contained in:
@@ -0,0 +1,12 @@
|
||||
DATALOADER:
|
||||
TRAIN_X:
|
||||
BATCH_SIZE: 256
|
||||
TEST:
|
||||
BATCH_SIZE: 256
|
||||
|
||||
OPTIM:
|
||||
NAME: "sgd"
|
||||
LR: 0.05
|
||||
STEPSIZE: [30]
|
||||
MAX_EPOCH: 30
|
||||
LR_SCHEDULER: "cosine"
|
||||
@@ -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,11 @@
|
||||
DATALOADER:
|
||||
TRAIN_X:
|
||||
BATCH_SIZE: 32
|
||||
TEST:
|
||||
BATCH_SIZE: 32
|
||||
|
||||
OPTIM:
|
||||
NAME: "sgd"
|
||||
LR: 0.002
|
||||
STEPSIZE: [20]
|
||||
MAX_EPOCH: 20
|
||||
@@ -0,0 +1,15 @@
|
||||
DATALOADER:
|
||||
TRAIN_X:
|
||||
BATCH_SIZE: 32
|
||||
TEST:
|
||||
BATCH_SIZE: 32
|
||||
|
||||
OPTIM:
|
||||
NAME: "sgd"
|
||||
LR: 0.0001
|
||||
STEPSIZE: [2]
|
||||
MAX_EPOCH: 2
|
||||
|
||||
TRAIN:
|
||||
PRINT_FREQ: 50
|
||||
COUNT_ITER: "train_u"
|
||||
Reference in New Issue
Block a user