diff options
| author | Cameron <cysmith1010@gmail.com> | 2016-11-06 22:35:10 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2016-11-06 22:35:10 -0700 |
| commit | 03e7c07720fb6a1ac166016fd1c33407fafee300 (patch) | |
| tree | 9a8d2704e8f205fa3ea2e0d3fffc534d8dec4413 | |
| parent | 0bbf6f551977adbe77750fe8a5515ad023f11277 (diff) | |
Update neural_style.py
| -rw-r--r-- | neural_style.py | 2 |
1 files changed, 1 insertions, 1 deletions
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)') |