asparagus/configs/default_finetune_seg.yaml at 02e6bfc75e740cf63dff75546d79434bef808a21 · lukasugar/asparagus · GitHub

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
defaults:
  - core/base@
  - hardware/1gpu40cpu@hardware
  - _self_

task:
checkpoint_run_id:
load_checkpoint_name: last.ckpt
root: base
stem: ${checkpoint_run_id}_${load_checkpoint_name}

model:
  net: ${model.seg_net}
  ckpt_every_n_epoch: 25

lightning:
  data_module: SegDataModule
  lightning_module: SegmentationModule

data:
  data_path: ${oc.env:ASPARAGUS_DATA}/${task}
  train_split: split_40_10_50
  test_split: TEST_40_10_50
  fold: 0

training:
  epochs: 1000
  batch_size: 2
  warmup_epochs: 50
  decoder_warmup_epochs: 50
  train_batches_per_epoch_per_device: 250
  val_batches_per_epoch_per_device: 50
  patch_size: [160, 160, 160]
  seed: ${random:0,1000000}
  check_val_every_n_epoch: 3
  accumulate_grad_batches: 1
  load_decoder: False
  repeat_stem_weights: True

transforms:
  cpu_tr_transforms: CPU_seg_train_transforms
  cpu_val_transforms: CPU_seg_val_transforms
  gpu_tr_transforms: GPU_all_train_transforms
  gpu_val_transforms:
  masking: False
  normalize: True

logger:
  progress_bar: True
  profile: False
  wandb_log_model: False
  wandb_logging: True
  wandb_entity: ${oc.env:WANDB_ENTITY, "team-asparagus"}
  wandb_project: ${task}
  mlflow_logging: False
  log_every_n_steps: 50
  log_images_every_n_epoch: 5
  log_to_stdout: True