diff --git a/neural_style_transfer.py b/neural_style_transfer.py index bfc70e6..8ea4233 100644 --- a/neural_style_transfer.py +++ b/neural_style_transfer.py @@ -144,8 +144,9 @@ def closure(): parser.add_argument("--style_img_name", type=str, help="style image name", default='vg_starry_night.jpg') parser.add_argument("--height", type=int, help="height of content and style images", default=400) - parser.add_argument("--content_weight", type=float, help="weight factor for content loss", default=1e5) - parser.add_argument("--style_weight", type=float, help="weight factor for style loss", default=3e4) + parser.add_argument('--style_weight', type=float, default=1e10, help='Weight for style loss.') + parser.add_argument('--content_weight', type=float, default=1e5, help='Weight for content loss.') + parser.add_argument("--tv_weight", type=float, help="weight factor for total variation loss", default=1e0) parser.add_argument("--optimizer", type=str, choices=['lbfgs', 'adam'], default='lbfgs')