Ahsen Khaliq commited on
Commit
d80e3bc
1 Parent(s): c4083ca

Update style_transfer_folder.py

Browse files
Files changed (1) hide show
  1. style_transfer_folder.py +1 -1
style_transfer_folder.py CHANGED
@@ -31,7 +31,7 @@ if __name__ == '__main__':
31
 
32
  parser.add_argument('--style_img_path', type=str, default=None, help='path to style image')
33
  parser.add_argument('--input_img_path', type=str, default=None, help='path to input image')
34
- parser.add_argument('--add_weight_index', type=int, default=10)
35
 
36
  parser.add_argument('--channel_multiplier', type=int, default=2)
37
  parser.add_argument('--outdir', type=str, default="")
31
 
32
  parser.add_argument('--style_img_path', type=str, default=None, help='path to style image')
33
  parser.add_argument('--input_img_path', type=str, default=None, help='path to input image')
34
+ parser.add_argument('--add_weight_index', type=int, default=6)
35
 
36
  parser.add_argument('--channel_multiplier', type=int, default=2)
37
  parser.add_argument('--outdir', type=str, default="")