config.yml

seed : 42
batch_size : 8
epochs : 1
lr : 0.001
weight_decay : 0.1
early_stopping_patience : 8

# spiking params are optional, if not specified, default values will be used
# models : resnet18, resnet18_spiking, slow_r50, vgg11_spiking, sew_resnet18_spiking
model : {
  type : temporal, # or temporal, or single_sample
  name : slow_r50,
  spiking_params : {
    neuron_model : plif,
    surrogate_function : sigmoid,
    n_samples : 10,
    step_mode : multi_step,
    backend : torch,
  }
}

dataset : {
  root_path : ./datasets/dataset_rgb,
  dvs_mode : False, # or False for RGB
  type : temporal, # either temporal, repeated, single_sample, prediction_temporal, prediction_single_sample, prediction_repeated
  img_width : 256,
  img_height : 450,
  evaluation_mode : standard, # or kfold
  n_folds : 5, # only used if evaluation_mode is kfold
  val_size : 0.15, # always used
  test_size : 0.15, # only used if evaluation_mode is standard
  timestep : 30, # only used if type is temporal
  overlap : 29, # only used if type is temporal
  repeats : 10, # only used if type is repeated
  n_frames_predictive_horizon : 150 # only used if type is prediction
}

wandb : {
  entity : mazurek,
  project : dvs_final_detection_seed,
  group : slow_r50_temporal
  }
checkpoint_path : checkpoints_lightning
dataset_save_dir : results_new_runs/datasets # save the test data for every fold (or entire run)
checkpoint_dir_to_restore : checkpoints_lightning/4d2ba12d-3762-412e-8648-609a789f2472