Spaces:
Configuration error
Configuration error
File size: 1,500 Bytes
3b40f46 |
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 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
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
|