PSPNet-tensorflow icon indicating copy to clipboard operation
PSPNet-tensorflow copied to clipboard

training from scratch

Open KodamaXYZ opened this issue 6 years ago • 1 comments

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

KodamaXYZ avatar Jan 29 '19 22:01 KodamaXYZ

Have you solved this issue? I'm having trouble training this right now!

AmeetR avatar Jun 03 '19 05:06 AmeetR