90 lines
2.3 KiB
YAML
90 lines
2.3 KiB
YAML
defaults:
|
|
- dset: dataset
|
|
|
|
path_experiment: "experiments/trained_model" #there should be a better way to do this
|
|
tensorboard_logs: "/scratch/work/molinee2/tensorboard_logs/unet_historical" #path with tensorboard
|
|
# Dataset related
|
|
fs: 44100 #default is 44100, better NEVER change
|
|
seg_len_s_train: 5 #length of the train (and val) segments in seconds
|
|
freq_inference: 10 #we do inference after * epochs
|
|
seg_len_s_test: 15 #lenum_test_segments: 10 #number of test segments (inferenced every epoch)
|
|
num_real_test_segments: 5 #number of real recordings inferenced every epoch
|
|
num_test_segments: 15
|
|
buffer_size: 1000 # buffer size for shuffling datasets (train and val)
|
|
# Dataset Augmentation
|
|
overlap: 0 #overlap when extracting audio segments, default is 0, augment if more data is needed
|
|
|
|
# Logging and printing, and does not impact training
|
|
#device: cuda
|
|
verbose: 0
|
|
use_tensorboard: True
|
|
use_soft_denoising: False
|
|
|
|
|
|
num_workers: 10
|
|
|
|
|
|
# Checkpointing, by default automatically load last checkpoint
|
|
checkpoint: true
|
|
continue_from: '' # Path the a checkpoint.th file to start from.
|
|
# this is not used in the name of the experiment!
|
|
# so use a dummy=something not to mixup experiments.
|
|
continue_best: false # continue from best, not last state if continue_from is set.
|
|
only_inference: false
|
|
|
|
# Optimization related
|
|
optim: adam
|
|
lr: 1e-4 #used
|
|
variable_lr: True
|
|
beta1: 0.5 #used
|
|
beta2: 0.9 #used
|
|
|
|
loss: "mae" #choose loss:
|
|
|
|
epochs: 73
|
|
batch_size: 16
|
|
val_take: -1
|
|
steps_per_epoch: 1000
|
|
|
|
|
|
sp:
|
|
method: "wiener"
|
|
|
|
#STFT parameteres
|
|
stft:
|
|
win_size: 2048 #STFT window size
|
|
hop_size: 512
|
|
|
|
#inference param
|
|
inference:
|
|
audio: None
|
|
# Models
|
|
model: unet # either demucs or dwave
|
|
unet:
|
|
activation: "elu"
|
|
use_csff: False
|
|
use_SAM: True
|
|
use_cam: False
|
|
use_fam: False
|
|
use_fencoding: True
|
|
use_tdf: False
|
|
use_alttdfs: False
|
|
num_tfc: 3
|
|
num_stages: 3
|
|
depth: 6
|
|
f_dim: 1025 #hardcoded, depends on the stft window
|
|
|
|
|
|
|
|
# Hydra config
|
|
hydra:
|
|
job:
|
|
config:
|
|
# configuration for the ${hydra.job.override_dirname} runtime variable
|
|
override_dirname:
|
|
kv_sep: '='
|
|
item_sep: ','
|
|
# Remove all paths, as the / in them would mess up things
|
|
exclude_keys: ['path_experiment',
|
|
'hydra.job_logging.handles.file.filename']
|