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

# data settings
data_args:
  path: data/har/processed
  # mode: fullts
  mode: fullts
  input_dims: 6
  subseq_size: 128
  train_stride: 32
  val_stride: 128

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

  recon_args:
    num_layers: 2
    hidden_dim: 64

  init_args:
    in_proj_kernel_size: 3
    in_proj_dilation: 1
    hidden_dim: 64
    emb_dim: 8
    init_noise_std: 0.1
    init_norm: false

  augmentation_args:
    # stretch_range: [0.95, 1.05]
    stretch_range: null
    n_init_samples: 2
    
    sample_right_boundary: 64
    add_recon_noise: false
    add_input_noise: false # what is this?
    add_init_noise: false
    init_noise_std: 0.1
    recon_noise_std: 0.25
    # sample_denom: 2
  

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