Skip to content
Snippets Groups Projects
Normal view Permalink
config_base.yaml 654 B
Newer Older
  • Learn to ignore specific revisions
    • Bobholamovic's avatar
    Hello Github
    Bobholamovic committed
    1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 
    # Basic configurations


# Data
# Common
dataset: OSCD
crop_size: 256
num_workers: 1
repeats: 1000


# Optimizer
optimizer: Adam
lr: 0.001
lr_mode: const
weight_decay: 0.0001
step: 2


# Training related
batch_size: 32
    Bobholamovic's avatar
    Fix extra channel in a saved image  
    Bobholamovic committed
    22 
    num_epochs: 40
    Bobholamovic's avatar
    Hello Github
    Bobholamovic committed
    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 
    resume: ''
load_optim: True
anew: False
trace_freq: 1
device: cuda
metrics: 'F1Score+Accuracy+Recall+Precision'


# Experiment
exp_dir: ../exp/
out_dir: ''
# tag: ''
# suffix: ''
# DO NOT specify exp-config term
save_on: False
log_off: False
suffix_off: False


# Criterion
criterion: NLL
weights: 
  - 1.0   # Weight of no-change class
  - 10.0   # Weight of change class


# Model
model: siamunet_conc
num_feats_in: 13