-
Notifications
You must be signed in to change notification settings - Fork 119
Open
Description
Hi hellochick, i love your model. i would like to ask how many steps did you train in order to get to the model added in the repository?
i would like to train the same amount of steps.
is it 60001?
BATCH_SIZE = 2
DATA_DIRECTORY = './data/cityscapes_dataset/cityscape/'
DATA_LIST_PATH = './list/cityscapes_train_list.txt'
IGNORE_LABEL = 255
INPUT_SIZE = '713,713'
LEARNING_RATE = 1e-3
MOMENTUM = 0.9
NUM_CLASSES = 19
NUM_STEPS = 60001
POWER = 0.9
RANDOM_SEED = 1234
WEIGHT_DECAY = 0.0001
RESTORE_FROM = './'
SNAPSHOT_DIR = './model/'
SAVE_NUM_IMAGES = 4
SAVE_PRED_EVERY = 20
Metadata
Metadata
Assignees
Labels
No labels