Spaces:
Configuration error
Configuration error
seed: -1 | |
use_wandb: False | |
wandb_project: project_name # enter your project name here | |
wandb_entity: project_entity # enter your entity here | |
device: cuda | |
log_images_freq: 50 | |
# dataset configuration | |
results_folder: "results" | |
resize_input: 512 | |
d_divisible_crops: 1 | |
source_image_every: 75 | |
crops_min_cover: 0.85 | |
# input augmentations | |
flip_p: 0.5 | |
jitter_p: 0.1 | |
scale_min: 0.8 | |
scale_max: 1.2 | |
# text configuration | |
#bootstrap_text: horse # this prompt is not composed with any augmentations at the moment | |
bootstrap_scheduler: linear # linear | exponential | none | |
text_criterion: spherical # spherical | cosine | |
# loss configuration | |
bootstrap_epoch: -1 | |
lambda_bootstrap: 10 | |
lambda_bootstrap_min: 0 | |
bootstrapping_min_cover: 1 | |
lambda_structure: 2 | |
lambda_screen: 1 | |
lambda_sparsity: 6 # lambda_sparsity * ( lambda_alpha_l0 * L0_loss + lambda_alpha_l1 * L1_loss ) | |
lambda_alpha_l0: 0.01 | |
lambda_alpha_l1: 0.01 | |
lambda_composition: 1 | |
use_negative_bootstrap: False | |
# CLIP handling configuration | |
n_aug: 16 # number of augmentations to be applied before CLIP | |
clip_model_name: "ViT-B/32" # ViT-B/16 | ViT-B/32 | ViT-L/14 | |
relevancy_num_layers: 10 | |
# CLIP augmentations | |
clip_affine_transform_fill: 1 | |
# training configuration | |
n_epochs: 1000 | |
scheduler_policy: exponential # [exponential| linear | step | plateau | cosine | none] | |
gamma: 0.99 | |
min_lr: 0.00001 | |
lr: 0.0025 | |
optimizer: madgrad # [madgrad | adam | radam | rmsprop | sgd] | |
# CNN backbone configuration | |
skip_n33d: 128 | |
skip_n33u: 128 | |
skip_n11: 4 | |
num_scales: 7 | |