base_config: configs/base/base.yaml
model_type: pulse
# device: cuda:0

# data settings
data_args:
  path: data/ppg/processed
  mode: subseq
  input_dims: 1
  subseq_size: 960
  train_stride: 500
  val_stride: 480
  downsample_factor: 4

model_args:
  time_vary_args:
    pool_denom: 1
    tv_dim: 1
    include: true
    tv_kernel_size: 3

  recon_args:
    num_layers: 4
    hidden_dim: 64

  init_args:
    in_proj_kernel_size: 5
    in_proj_dilation: 2
    hidden_dim: 64
    emb_dim: 8

  augmentation_args:
    n_init_samples: 2
    sample_right_boundary: 480
  

# optimizer settings
training_args:
  epochs: 100
  batch_size: 128
  eval_batch_size: 128
  lr: 0.0001
  save_every_n: 100
