import models.RevNet as RevNet import os import scheduler os.environ["CUDA_VISIBLE_DEVICES"] = '0' BATCH_SIZE = 128 N_WORKERS = 4 EPOCHS = 300 PATIENCE = -1 LR = 0.1 W_DECAY = 5e-4#2e-4 MOMENTUM = 0.9#0.9 # LRS = [0.1,0.01,0.001] LRS = [0.1, 0.02, 0.004, 0.0008] LR_EPOCH = [60, 120, 160, 200] MODEL = RevNet.RevNet48GN(group_norm=8) # MODEL = RevNet.RevNet104() def SCHEDULER(optimizer): return scheduler.SimpleScheduler(optimizer, LR_EPOCH, LRS) FOLD = "revnet48_gn"
import models.RevNet as RevNet
import os
import scheduler
os.environ["CUDA_VISIBLE_DEVICES"] = '3'
BATCH_SIZE = 128
N_WORKERS = 4
EPOCHS = 300
PATIENCE = -1
LR = 0.1
W_DECAY = 5e-4 #2e-4
MOMENTUM = 0.9 #0.9
# LRS = [0.1,0.01,0.001]
LRS = [0.1, 0.02, 0.004, 0.0008]
LR_EPOCH = [60, 120, 160, 200]
exp = 8
MODEL = RevNet.RevNet104([64 * exp, 128 * exp, 256 * exp, 512 * exp])
# MODEL = RevNet.RevNet104()
def SCHEDULER(optimizer):
return scheduler.SimpleScheduler(optimizer, LR_EPOCH, LRS)
FOLD = "revnet104_ssc"
import models.RevNet as RevNet import os os.environ["CUDA_VISIBLE_DEVICES"] = '0' # 3 > c2 # 2 > c1 import scheduler BATCH_SIZE = 128 N_WORKERS = 4 EPOCHS = 350 #PATIENCE = 30 # PATIENCE = 5 PATIENCE = 30 LR = 0.1 W_DECAY = 5e-4#2e-4 MOMENTUM = 0.9#0.9 LRS = [0.1, 0.01, 0.001] def SCHEDULER(optimizer): return scheduler.MoreSimpleScheduler(optimizer, LRS) MODEL = RevNet.RevNet102bsc() FOLD = "revnet102"
import scheduler
import torch.optim as optim
BATCH_SIZE = 128
N_WORKERS = 4
EPOCHS = 300
#PATIENCE = 30
# PATIENCE = 5
PATIENCE = -1
LR = 0.1
W_DECAY = 5e-4 #2e-4
MOMENTUM = 0.9 #0.9
# LRS = [1,0.1, 0.01, 0.001, 0.0001] # fold = revnet18_morelr
# LRS = [0.0001, 0.001, 0.01, 0.1, 0.01, 0.001, 0.0001]
LRS = [0.1, 0.02, 0.004, 0.0008]
LR_EPOCH = [60, 120, 160, 200]
def SCHEDULER(optimizer):
#return scheduler.MoreSimpleScheduler(optimizer, LRS)
return scheduler.SimpleScheduler(optimizer, LR_EPOCH, LRS)
# return optim.lr_scheduler.CosineAnnealingLR(optimizer, 12, eta_min=0, last_epoch=-1)
MODEL = RevNet.RevNet18()
FOLD = "revnet18_callr"
import models.RevNet as RevNet
import os
import scheduler
os.environ["CUDA_VISIBLE_DEVICES"] = '3'
BATCH_SIZE = 128
N_WORKERS = 4
EPOCHS = 300
PATIENCE = -1
LR = 0.1
W_DECAY = 5e-4 #2e-4
MOMENTUM = 0.9 #0.9
# LRS = [0.1,0.01,0.001]
LRS = [0.1, 0.02, 0.004, 0.0008]
LR_EPOCH = [60, 120, 160, 200]
MODEL = RevNet.RevNet98GN(group_norm=16)
# MODEL = RevNet.RevNet104()
def SCHEDULER(optimizer):
return scheduler.SimpleScheduler(optimizer, LR_EPOCH, LRS)
FOLD = "revnet98_gn16"
import models.RevNet as RevNet
import os
os.environ["CUDA_VISIBLE_DEVICES"] = '1'
import scheduler
BATCH_SIZE = 128
N_WORKERS = 4
EPOCHS = 300
PATIENCE = -1
LR = 0.1
W_DECAY = 5e-4 #2e-4
MOMENTUM = 0.9 #0.9
LRS = [0.1, 0.02, 0.004, 0.0008]
LR_EPOCH = [60, 120, 160, 200]
def SCHEDULER(optimizer):
return scheduler.SimpleScheduler(optimizer, LR_EPOCH, LRS)
MODEL = RevNet.RevNet98(num_classes=100)
FOLD = "revnet98_cifar100"
import models.RevNet as RevNet
import os
import scheduler
os.environ["CUDA_VISIBLE_DEVICES"] = '0'
BATCH_SIZE = 128
N_WORKERS = 4
EPOCHS = 300
PATIENCE = -1
LR = 0.1
W_DECAY = 5e-4 #2e-4
MOMENTUM = 0.9 #0.9
# LRS = [0.1,0.01,0.001]
LRS = [0.1, 0.02, 0.004, 0.0008]
LR_EPOCH = [60, 120, 160, 200]
MODEL = RevNet.RevNet34()
def SCHEDULER(optimizer):
return scheduler.SimpleScheduler(optimizer, LR_EPOCH, LRS)
FOLD = "revnet34_ssc"
import models.RevNet as RevNet
import os
import scheduler
os.environ["CUDA_VISIBLE_DEVICES"] = '3'
BATCH_SIZE = 128
N_WORKERS = 4
EPOCHS = 600
PATIENCE = -1
LR = 0.1
W_DECAY = 5e-4 #2e-4
MOMENTUM = 0.9 #0.9
# LRS = [0.1,0.01,0.001]
LRS = [0.1, 0.02, 0.004, 0.0008]
LR_EPOCH = [60 * 2, 120 * 2, 160 * 2, 200 * 2]
MODEL = RevNet.RevNet98()
# MODEL = RevNet.RevNet104()
def SCHEDULER(optimizer):
return scheduler.SimpleScheduler(optimizer, LR_EPOCH, LRS)
FOLD = "revnet98_longer_training"
import models.RevNet as RevNet
import os
os.environ["CUDA_VISIBLE_DEVICES"] = '2'
import scheduler
BATCH_SIZE = 128
N_WORKERS = 4
EPOCHS = 300
PATIENCE = -1
LR = 0.1
W_DECAY = 5e-4 #2e-4
MOMENTUM = 0.9 #0.9
LRS = [0.1, 0.02, 0.004, 0.0008]
LR_EPOCH = [60, 120, 160, 200]
def SCHEDULER(optimizer):
return scheduler.SimpleScheduler(optimizer, LR_EPOCH, LRS)
MODEL = RevNet.RevNet162(num_classes=100)
FOLD = "revnet162_cifar100"
import models.RevNet as RevNet
import os
import scheduler
os.environ["CUDA_VISIBLE_DEVICES"] = '0'
BATCH_SIZE = 128
N_WORKERS = 4
EPOCHS = 300
PATIENCE = -1
LR = 0.1
W_DECAY = 5e-4 #2e-4
MOMENTUM = 0.9 #0.9
# LRS = [0.1,0.01,0.001]
LRS = [0.1, 0.02, 0.004, 0.0008]
LR_EPOCH = [60, 120, 160, 200]
MODEL = RevNet.RevNet48()
# MODEL = RevNet.RevNet104()
def SCHEDULER(optimizer):
return scheduler.SimpleScheduler(optimizer, LR_EPOCH, LRS)
FOLD = "revnet48"
import models.RevNet as RevNet import os import scheduler os.environ["CUDA_VISIBLE_DEVICES"] = '3' BATCH_SIZE = 128 N_WORKERS = 4 EPOCHS = 600 PATIENCE = -1 LR = 0.1 W_DECAY = 5e-4#2e-4 MOMENTUM = 0.9#0.9 # LRS = [0.1,0.01,0.001] LRS = [0.1, 0.02, 0.004, 0.0008] LR_EPOCH = [60*2, 120*2, 160*2, 200*2] MODEL = RevNet.RevNet162() # MODEL = RevNet.RevNet104() def SCHEDULER(optimizer): return scheduler.SimpleScheduler(optimizer, LR_EPOCH, LRS) FOLD = "revnet162_longer_training"
import models.RevNet as RevNet import os os.environ["CUDA_VISIBLE_DEVICES"] = '2' import scheduler BATCH_SIZE = 128 N_WORKERS = 4 EPOCHS = 300 PATIENCE = -1 LR = 0.1 W_DECAY = 5e-4#2e-4 MOMENTUM = 0.9#0.9 LRS = [0.1, 0.02, 0.004, 0.0008] LR_EPOCH = [60, 120, 160, 200] def SCHEDULER(optimizer): return scheduler.SimpleScheduler(optimizer, LR_EPOCH, LRS) MODEL = RevNet.RevNet34(num_classes=100) FOLD = "revnet34_cifar100"
