From 03e7c07720fb6a1ac166016fd1c33407fafee300 Mon Sep 17 00:00:00 2001 From: Cameron Date: Sun, 6 Nov 2016 22:35:10 -0700 Subject: Update neural_style.py --- neural_style.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/neural_style.py b/neural_style.py index 59491d6..3156bdd 100644 --- a/neural_style.py +++ b/neural_style.py @@ -81,7 +81,7 @@ def parse_args(): default=['relu1_1', 'relu2_1', 'relu3_1', 'relu4_1', 'relu5_1'], help='VGG19 layers used for the style image. (default: %(default)s)') - parser.add_argument('--content_layer_weights', type=float, + parser.add_argument('--content_layer_weights', nargs='+', type=float, default=[1.0], help='Contributions (weights) of each content layer to loss. (default: %(default)s)') -- cgit v1.2.3-70-g09d2