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


# data settings
data_args:
  path: data/sleep_eeg/processed/healthy
  mode: fullts
  input_dims: 2
  subseq_size: 1000
  train_stride: 350
  val_stride: 1000
  downsample_factor: 3
  # include_channels: [0]
  p_train: 0.5
  p_val: 0.5


model_args:
  time_vary_args:
    include: true
    pool_denom: 2 # 5: 0.8485 # 10: 0.8478
    tv_dim: 2
    tv_kernel_size: 3

  recon_args:
    num_layers: 2
    hidden_dim: 64

  init_args:
    in_proj_kernel_size: 3
    in_proj_dilation: 2
    hidden_dim: 64
    emb_dim: 12

  augmentation_args:
    n_init_samples: 1
    sample_right_boundary: 500
  

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