File size: 554 Bytes
a0bcaae
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
## Architechture
lpips_type = 'alex'
first_inv_type = 'w+'#'w+'
optim_type = 'adam'

## Locality regularization
latent_ball_num_of_samples = 1
locality_regularization_interval = 1
use_locality_regularization = False
regulizer_l2_lambda = 0.1
regulizer_lpips_lambda = 0.1
regulizer_alpha = 30

## Loss
pt_l2_lambda = 1
pt_lpips_lambda = 1

## Steps
LPIPS_value_threshold = 0.04
max_pti_steps = 350
first_inv_steps = 450
max_images_to_invert = 30

## Optimization
pti_learning_rate = 5e-4
first_inv_lr = 8e-3
train_batch_size = 1
use_last_w_pivots = False