Python ResNet.ResNet34示例

示例#1

import models.ResNet as ResNet
import os
import scheduler
os.environ["CUDA_VISIBLE_DEVICES"] = '0'

BATCH_SIZE = 128
N_WORKERS = 2

EPOCHS = 1
PATIENCE = -1

LR = 0.1
W_DECAY = 5e-4  #2e-4
MOMENTUM = 0.9  #0.9

LRS = [0.1, 0.01, 0.001]

MODEL = ResNet.ResNet34()


def SCHEDULER(optimizer):
    return scheduler.MoreSimpleScheduler(optimizer, LRS)


FOLD = "check_resnet34"

示例#2

文件： config_cifar100_resnet34.py 项目： Myyyr/classification

import models.ResNet as ResNet
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 = ResNet.ResNet34(num_classes=100)

FOLD = "resnet34_cifar100_ssc"