PTI/configs/hyperparameters.py

## Architechture
lpips_type = "alex"
first_inv_type = "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
use_mask = True
pt_l2_lambda = 0.7
pt_lpips_lambda = 1
color_transfer_lambda = 0  # 1e6
id_lambda = 1

## Steps
LPIPS_value_threshold = 0.01 # 0.06
max_pti_steps = 350
first_inv_steps = 450
max_images_to_invert = 10

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