diff --git a/recipes/intel_ndns/spiking_fullsubnet_v2/default.yaml b/recipes/intel_ndns/spiking_fullsubnet_v2/conf/default.yaml similarity index 100% rename from recipes/intel_ndns/spiking_fullsubnet_v2/default.yaml rename to recipes/intel_ndns/spiking_fullsubnet_v2/conf/default.yaml diff --git a/recipes/intel_ndns/spiking_fullsubnet_v2/eval_middle.yaml b/recipes/intel_ndns/spiking_fullsubnet_v2/conf/eval_middle.yaml similarity index 100% rename from recipes/intel_ndns/spiking_fullsubnet_v2/eval_middle.yaml rename to recipes/intel_ndns/spiking_fullsubnet_v2/conf/eval_middle.yaml diff --git a/recipes/intel_ndns/spiking_fullsubnet_v2/middle_8-32-64_311.yaml b/recipes/intel_ndns/spiking_fullsubnet_v2/conf/middle-model__partition-0-32-128-256__grouping-8-32-64__deep-filtering-3-1-1.yaml similarity index 100% rename from recipes/intel_ndns/spiking_fullsubnet_v2/middle_8-32-64_311.yaml rename to recipes/intel_ndns/spiking_fullsubnet_v2/conf/middle-model__partition-0-32-128-256__grouping-8-32-64__deep-filtering-3-1-1.yaml diff --git a/recipes/intel_ndns/spiking_fullsubnet_v2/middle_8-32-64_531.yaml b/recipes/intel_ndns/spiking_fullsubnet_v2/conf/middle-model__partition-0-32-128-256__grouping-8-32-64__deep-filtering-5-3-1.yaml similarity index 91% rename from recipes/intel_ndns/spiking_fullsubnet_v2/middle_8-32-64_531.yaml rename to recipes/intel_ndns/spiking_fullsubnet_v2/conf/middle-model__partition-0-32-128-256__grouping-8-32-64__deep-filtering-5-3-1.yaml index a104ae2..e68ec87 100644 --- a/recipes/intel_ndns/spiking_fullsubnet_v2/middle_8-32-64_531.yaml +++ b/recipes/intel_ndns/spiking_fullsubnet_v2/conf/middle-model__partition-0-32-128-256__grouping-8-32-64__deep-filtering-5-3-1.yaml @@ -1,5 +1,5 @@ trainer: - output_dir: exp/middle_8-32-64_311 + output_dir: exp/middle-model__partition-0-32-128-256__grouping-8-32-64__deep-filtering-5-3-1 do_train: True resume_from_checkpoint: "no" per_device_train_batch_size: 64 diff --git a/recipes/intel_ndns/spiking_fullsubnet_v2/middle_533.yaml b/recipes/intel_ndns/spiking_fullsubnet_v2/conf/middle_533.yaml similarity index 100% rename from recipes/intel_ndns/spiking_fullsubnet_v2/middle_533.yaml rename to recipes/intel_ndns/spiking_fullsubnet_v2/conf/middle_533.yaml diff --git a/recipes/intel_ndns/spiking_fullsubnet_v2/middle_711.yaml b/recipes/intel_ndns/spiking_fullsubnet_v2/conf/middle_711.yaml similarity index 100% rename from recipes/intel_ndns/spiking_fullsubnet_v2/middle_711.yaml rename to recipes/intel_ndns/spiking_fullsubnet_v2/conf/middle_711.yaml diff --git a/recipes/intel_ndns/spiking_fullsubnet_v2/conf/small-model__partition-0-32-128-256__grouping-8-32-64__deep-filtering-5-3-1.yaml b/recipes/intel_ndns/spiking_fullsubnet_v2/conf/small-model__partition-0-32-128-256__grouping-8-32-64__deep-filtering-5-3-1.yaml new file mode 100644 index 0000000..83d36b5 --- /dev/null +++ b/recipes/intel_ndns/spiking_fullsubnet_v2/conf/small-model__partition-0-32-128-256__grouping-8-32-64__deep-filtering-5-3-1.yaml @@ -0,0 +1,47 @@ +trainer: + output_dir: exp/small-model__partition-0-32-128-256__grouping-8-32-64__deep-filtering-5-3-1 + do_train: True + resume_from_checkpoint: "no" + per_device_train_batch_size: 64 + eval_batch_size: 8 + dataloader_num_workers: 8 + num_train_epochs: 200 + max_grad_norm: 10 + save_epoch_interval: 5 + eval_epoch_interval: 5 + early_stopping_patience: 20 + save_total_limit: 20 + +model: + n_fft: 512 + hop_length: 128 + win_length: 512 + fdrc: 0.5 + fb_input_size: 64 + fb_hidden_size: 240 + fb_num_layers: 2 + fb_proj_size: 64 + fb_output_activate_function: null + sb_hidden_size: 160 + sb_num_layers: 2 + freq_cutoffs: [0, 32, 128, 256] + df_orders: [5, 3, 1] + center_freq_sizes: [8, 32, 64] + neighbor_freq_sizes: [15, 15, 15] + use_pre_layer_norm_fb: True + use_pre_layer_norm_sb: True + bn: True + shared_weights: True + sequence_model: "GSN" + num_spks: 1 + +train_dataset: + root: /datasets/datasets_fullband/training_set/ + limit: False + offset: 0 + +eval_dataset: + root: /datasets/datasets_fullband/validation_set/ + train: False + limit: 200 + offset: 0