Spaces:
Runtime error
Runtime error
File size: 1,434 Bytes
35188e4 |
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 |
# augmentations
use_copy_paste: false
scale_range: [ 0.1, 1.0 ]
repeat_image: false
# base directories
dir_ckpt: "/users/gyungin/selfmask/ckpt" # "/work/gyungin/selfmask/ckpt"
dir_dataset: "/scratch/shared/beegfs/gyungin/datasets"
# clustering
k: [2, 3, 4]
clustering_mode: "spectral"
use_gpu: true # if you want to use gpu-accelerated code for clustering
scale_factor: 2 # "how much you want to upsample encoder features before clustering"
# dataset
dataset_name: "duts"
use_pseudo_masks: true
train_image_size: 224
eval_image_size: 224
n_percent: 100
n_copy_pastes: null
pseudo_masks_fp: "/users/gyungin/selfmask/datasets/swav_mocov2_dino_p16_k234.json"
# dataloader:
batch_size: 8
num_workers: 4
pin_memory: true
# networks
abs_2d_pe_init: false
arch: "vit_small"
lateral_connection: false
learnable_pixel_decoder: false # if False, use the bilinear interpolation
use_binary_classifier: true # if True, use a binary classifier to get an objectness for each query from transformer decoder
n_decoder_layers: 6
n_queries: 20
num_layers: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11]
patch_size: 8
training_method: "dino" # "supervised", "deit", "dino", "mocov2", "swav"
# objective
loss_every_decoder_layer: true
weight_dice_loss: 1.0
weight_focal_loss: 0.0
# optimizer
lr: 0.000006 # default: 0.00006
lr_warmup_duration: 0 # 5
momentum: 0.9
n_epochs: 12
weight_decay: 0.01
optimizer_type: "adamw"
# validation
benchmarks: null |