From 5e4199d8d12f953fc2907fcee1845a8952fe75df Mon Sep 17 00:00:00 2001 From: balajicoding <164991652+balajicoding@users.noreply.github.com> Date: Mon, 21 Apr 2025 20:25:22 +0530 Subject: [PATCH] Update neural_style_transfer.py --- neural_style_transfer.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) 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')