PSPNet-tensorflow
PSPNet-tensorflow copied to clipboard
training from scratch
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
Have you solved this issue? I'm having trouble training this right now!