Kush26 commited on
Commit
ab248ac
·
verified ·
1 Parent(s): f06d10c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -130,7 +130,7 @@ def stylize_image(content_image: Image.Image):
130
  generated_img = new_content_img.clone().requires_grad_(True).to(device)
131
 
132
  # 3. Setup optimizer for the generated image
133
- lr = 0.02
134
  optimizer = optim.Adam([generated_img], lr=lr)
135
 
136
  # 4. Run optimization loop
 
130
  generated_img = new_content_img.clone().requires_grad_(True).to(device)
131
 
132
  # 3. Setup optimizer for the generated image
133
+ lr = 0.002
134
  optimizer = optim.Adam([generated_img], lr=lr)
135
 
136
  # 4. Run optimization loop