Python RevNet.RevNet162の例

コード例 #1

ファイル: config_cifar100_revnet162.py プロジェクト: Myyyr/classification

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"

コード例 #2

ファイル: config_revnet162.py プロジェクト: Myyyr/classification

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"