base_config: configs/base/base.yaml
model_type: pulse
# device: cuda:1
load_from_checkpoint: null

# data settings
data_args:
  path: data/ecg/processed
  mode: subseq
  # mode: subseq
  input_dims: 2
  subseq_size: 500
  train_stride: 250
  val_stride: 250
  downsample_factor: 5
  p_train: 0.5
  p_val: 0.5


model_args:
  time_vary_args:
    include: true
    pool_denom: 100 # 25: 0.77, 50: 0.823, 100: 0.855
    tv_dim: 1
    tv_kernel_size: 2

  recon_args:
    num_layers: 2
    hidden_dim: 64

  init_args:
    in_proj_kernel_size: 11
    in_proj_dilation: 2
    hidden_dim: 128
    emb_dim: 8

  augmentation_args:
    n_init_samples: 2
    sample_right_boundary: 250
  

# optimizer settings
training_args:
  epochs: 100
  batch_size: 64
  eval_batch_size: 64
  lr: 0.001 # 0.001
  save_every_n: 100
