diff --git a/TensorFlow/contrib/cv/FAST-STYLE-TRANSFER_ID2056_for_TensorFlow/style.py b/TensorFlow/contrib/cv/FAST-STYLE-TRANSFER_ID2056_for_TensorFlow/style.py index cc2998bfe8e9b551088b8d149c6e6865607f28e9..93a1a2dce41d6dfd9be73c9fe5d22d052832b4ca 100644 --- a/TensorFlow/contrib/cv/FAST-STYLE-TRANSFER_ID2056_for_TensorFlow/style.py +++ b/TensorFlow/contrib/cv/FAST-STYLE-TRANSFER_ID2056_for_TensorFlow/style.py @@ -27,8 +27,8 @@ CONTENT_WEIGHT = 7.5e0 STYLE_WEIGHT = 1e2 TV_WEIGHT = 2e2 -LEARNING_RATE = 1e-3 -NUM_EPOCHS = 2 +LEARNING_RATE = 1.8e-3 +NUM_EPOCHS = 12 CHECKPOINT_DIR = 'checkpoints' CHECKPOINT_ITERATIONS = 2000 #VGG_PATH = '/public/home/nanrh123/pyterm/fast-style-transfer-master/data/imagenet-vgg-verydeep-19.mat'