spatial-augmentations/configs/experiment/domain7_hparam_baseline.yaml at main · BoevaLab/spatial-augmentations · 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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
# @package _global_

# to execute this experiment run:
# python src/train_domain.py -m hparams_search=domain7_baseline experiment=domain7_hparam_baseline

defaults:
  - override /data: spatial_omics
  - override /model: bgrl_domain
  - override /trainer: gpu

# all parameters below will be merged with parameters from default configurations set above
# this allows you to overwrite only specified parameters

task_name: "domain7_hparam_baseline"
tags: ["domain7", "hparam", "baseline"]
validate: True

data:
  data_dir: ${paths.data_dir}domain/raw_7/
  processed_dir: ${paths.data_dir}domain/processed_7/
  batch_size: 8
  num_workers: 4
  pin_memory: False
  min_cells: 2
  min_genes: 10
  augmentation_mode: "baseline"
  lambda_param: 0.1
  sigma_param: 0.1
  n_pca_components: 50
  graph_method: "knn"
  n_neighbors: 10
  redo_preprocess: False

model:
  net:
    encoder:
      input_size: 50
      hidden_size: 32
      output_size: 16
      dropout: 0.5
    projector:
      input_size: ${model.net.encoder.output_size}
      output_size: ${model.net.encoder.output_size}
      hidden_size: 32

  optimizer:
    _partial_: true
    lr: 0.005
    weight_decay: 1e-4

  scheduler:
    _partial_: true
    T_max: 25974
    eta_min: 1e-7

  compile: false
  augmentation_mode: "baseline"
  mm: 0.99
  warmup_steps: 100
  total_steps: 25974
  drop_edge_p1: 0.3
  drop_edge_p2: 0.3
  drop_feat_p1: 0.3
  drop_feat_p2: 0.3
  mu: 0.5
  spatial_regularization_strength: 0.0
  node_subset_sz: 5000
  p_lambda: 0.7
  processed_dir: ${paths.data_dir}domain/processed_7/

trainer:
  min_epochs: 1
  max_epochs: 999
  log_every_n_steps: 4
  check_val_every_n_epoch: 9999
  precision: 16
  accumulate_grad_batches: 4