asparagus/configs/default_finetune_reg.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
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.cls_net}
  ckpt_every_n_epoch: 25

lightning:
  data_module: ClsRegDataModule
  lightning_module: RegressionModule

data:
  data_path: ${oc.env:ASPARAGUS_DATA}/${task}
  train_split: split_75_15_10
  fold: 0

training:
  epochs: 50
  batch_size: 2
  limit_train_batches: 250
  limit_val_batches: 50
  target_size: [128, 128, 128]
  seed: ${random:0,1000000}
  accumulate_grad_batches: 1
  warmup_epochs: 0
  check_val_every_n_epoch: 1
  load_decoder: True

transforms:
  cpu_tr_transforms: CPU_clsreg_train_transforms_crop
  cpu_val_transforms: CPU_clsreg_val_test_transforms_crop
  gpu_tr_transforms: GPU_all_train_transforms
  gpu_val_transforms:
  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: Finetune
  mlflow_logging: False
  log_every_n_steps: 250
  log_images_every_n_epoch: 5
  log_to_stdout: True