Index A | B | C | D | E | F | G | H | I | L | M | N | P | Q | R | S | T | U | V | X A ANNBase (class in src.model.ArtificialModel) ATan (class in src.model.SurrogateGradient) audio_reconstruction() (src.data.DatasetManager.DatasetManager method) audio_to_tensor() (src.data.DatasetManager.DatasetManager method) B backward() (src.model.SurrogateGradient.ATan static method) (src.model.SurrogateGradient.PiecewiseLinear static method) (src.model.SurrogateGradient.SigmoidDerivative static method) (src.model.SurrogateGradient.SuperSpike static method) C clamp_neuron_parameters() (src.model.SpikingLayer.MPNeuronModel method) (src.model.SpikingLayer.NeuronModel method) (src.model.SpikingModel.SNNBase method) CNN (class in src.model.ArtificialModel) compute_composite_objective_eval_metrics() (src.evaluation.EvaluationManager.EvaluationManager method) compute_dnsmos_objective_eval_metrics() (src.evaluation.EvaluationManager.EvaluationManager method) compute_dnsmos_score() (src.evaluation.EvaluationManager.EvaluationManager method) compute_evaluation_metrics() (src.evaluation.EvaluationManager.EvaluationManager method) compute_mean_dnsmos() (src.evaluation.EvaluationManager.EvaluationManager method) compute_objective_eval_metrics() (src.evaluation.EvaluationManager.EvaluationManager method) compute_representation_fn() (src.data.DatasetManager.DatasetManager method) compute_stft() (src.stft.StftManager.StftManager method) Conv2dBlock (class in src.model.ArtificialBlock) CSNN (class in src.model.SpikingModel) D DatasetManager (class in src.data.DatasetManager) E EvaluationManager (class in src.evaluation.EvaluationManager) experiment_tags() (src.model.SpeechEnhancer.SpeechEnhancer method) F FCSNN (class in src.model.SpikingModel) forward() (src.model.ArtificialBlock.Conv2dBlock method) (src.model.ArtificialBlock.ResConv2dBlock method) (src.model.ArtificialModel.CNN method) (src.model.ArtificialModel.ResBottleneckUNet method) (src.model.ArtificialModel.UNet method) (src.model.LossManager.LossManager method) (src.model.SpikingBlock.ReadoutConv2dBlock method) (src.model.SpikingBlock.ReadoutFCBlock method) (src.model.SpikingBlock.SpikingConv2dBlock method) (src.model.SpikingBlock.SpikingFCBlock method) (src.model.SpikingBlock.SpikingResConv2dBlock method) (src.model.SpikingLayer.Linear1d method) (src.model.SpikingLayer.MPNeuronModel method) (src.model.SpikingLayer.NeuronModel method) (src.model.SpikingLayer.Recurrent1d method) (src.model.SpikingLayer.Recurrent2d method) (src.model.SpikingLayer.ScaleLayer method) (src.model.SpikingLayer.Upsampling2d method) (src.model.SpikingModel.CSNN method) (src.model.SpikingModel.FCSNN method) (src.model.SpikingModel.ResBottleneckUNetSNN method) (src.model.SpikingModel.UNetSNN method) (src.model.SurrogateGradient.ATan static method) (src.model.SurrogateGradient.PiecewiseLinear static method) (src.model.SurrogateGradient.SigmoidDerivative static method) (src.model.SurrogateGradient.SuperSpike static method) G get_coefficients() (src.data.DatasetManager.DatasetManager method) get_params_dict() (src.model.SpeechEnhancer.SpeechEnhancer method) get_transform_info() (src.data.TransformManager.TransformManager method) get_weights() (src.data.DatasetManager.DatasetManager method) H huber_loss() (src.model.LossManager.LossManager method) I init_neuron_parameters() (src.model.SpikingBlock.ReadoutConv2dBlock method) (src.model.SpikingBlock.ReadoutFCBlock method) (src.model.SpikingBlock.SpikingConv2dBlock method) (src.model.SpikingBlock.SpikingFCBlock method) (src.model.SpikingBlock.SpikingResConv2dBlock method) (src.model.SpikingLayer.MPNeuronModel method) (src.model.SpikingLayer.NeuronModel method) init_rec() (src.model.ArtificialModel.ANNBase method) (src.model.SpikingModel.SNNBase method) init_recurrent_weights() (src.model.SpikingBlock.SpikingConv2dBlock method) (src.model.SpikingBlock.SpikingFCBlock method) (src.model.SpikingBlock.SpikingResConv2dBlock method) init_state() (src.model.SpikingModel.SNNBase method) init_state_none() (src.model.SpikingModel.SNNBase method) init_weights() (src.model.ArtificialBlock.Conv2dBlock method) (src.model.ArtificialBlock.ResConv2dBlock method) (src.model.SpikingBlock.ReadoutConv2dBlock method) (src.model.SpikingBlock.ReadoutFCBlock method) (src.model.SpikingBlock.SpikingConv2dBlock method) (src.model.SpikingBlock.SpikingFCBlock method) (src.model.SpikingBlock.SpikingResConv2dBlock method) inverse_log() (src.data.TransformManager.TransformManager method) inverse_log10() (src.data.TransformManager.TransformManager method) inverse_log_power() (src.data.TransformManager.TransformManager method) inverse_maxabs() (src.data.TransformManager.TransformManager method) inverse_normalize() (src.data.TransformManager.TransformManager method) inverse_quantile_maxabs() (src.data.TransformManager.TransformManager method) inverse_shift() (src.data.TransformManager.TransformManager method) inverse_standardize() (src.data.TransformManager.TransformManager method) L l1_loss() (src.model.LossManager.LossManager method) LI1d (class in src.model.SpikingLayer) LI2d (class in src.model.SpikingLayer) LIF1d (class in src.model.SpikingLayer) LIF2d (class in src.model.SpikingLayer) Linear1d (class in src.model.SpikingLayer) load_audio() (src.data.DatasetManager.DatasetManager method) load_checkpoint() (src.model.TrainValidTestManager.TrainValidTestManager method) log() (src.data.TransformManager.TransformManager method) log10() (src.data.TransformManager.TransformManager method) log_power() (src.data.TransformManager.TransformManager method) LossManager (class in src.model.LossManager) lsd_loss() (src.model.LossManager.LossManager method) M maxabs() (src.data.TransformManager.TransformManager method) mini_batch (src.stft.StftManager.Stft property) mini_batch_dir (src.stft.StftManager.Stft property) MPNeuronModel (class in src.model.SpikingLayer) mse_loss() (src.model.LossManager.LossManager method) multi_step_LI() (src.model.SpikingLayer.LI1d method) (src.model.SpikingLayer.LI2d method) (src.model.SpikingLayer.MPNeuronModel method) multi_step_neuron() (src.model.SpikingLayer.LIF1d method) (src.model.SpikingLayer.LIF2d method) (src.model.SpikingLayer.NeuronModel method) N NeuronModel (class in src.model.SpikingLayer) normalize() (src.data.TransformManager.TransformManager method) P perceptual_metric() (src.model.LossManager.LossManager method) pesq_fn() (src.evaluation.EvaluationManager.EvaluationManager static method) (src.model.LossManager.LossManager method) PiecewiseLinear (class in src.model.SurrogateGradient) plot_ann_layers() (src.visualization.VisualizationManager.VisualizationManager method) plot_coefficients() (src.visualization.VisualizationManager.VisualizationManager method) plot_dist() (src.visualization.VisualizationManager.VisualizationManager method) plot_input_data() (src.model.SpeechEnhancer.SpeechEnhancer method) plot_loss() (src.visualization.VisualizationManager.VisualizationManager method) plot_membrane_potential() (src.visualization.VisualizationManager.VisualizationManager method) plot_output_data() (src.model.SpeechEnhancer.SpeechEnhancer method) plot_perceptual_metric() (src.visualization.VisualizationManager.VisualizationManager method) plot_raster() (src.visualization.VisualizationManager.VisualizationManager method) plot_signal() (src.visualization.VisualizationManager.VisualizationManager method) plot_signal_data() (src.visualization.VisualizationManager.VisualizationManager method) plot_snn_layers() (src.visualization.VisualizationManager.VisualizationManager method) plot_spectrogram() (src.visualization.VisualizationManager.VisualizationManager method) plot_weights() (src.model.SpeechEnhancer.SpeechEnhancer method) prepare_data_manager() (src.model.SpeechEnhancer.SpeechEnhancer method) prepare_model() (src.model.SpeechEnhancer.SpeechEnhancer method) prepare_optimizer() (src.model.TrainValidTestManager.TrainValidTestManager method) prepare_scheduler() (src.model.TrainValidTestManager.TrainValidTestManager method) prepare_valid_audio() (src.data.DatasetManager.DatasetManager method) Q quantile_maxabs() (src.data.TransformManager.TransformManager method) R ReadoutConv2dBlock (class in src.model.SpikingBlock) ReadoutFCBlock (class in src.model.SpikingBlock) Recurrent1d (class in src.model.SpikingLayer) Recurrent2d (class in src.model.SpikingLayer) res_skip_connection() (src.model.ArtificialBlock.ResConv2dBlock method) ResBottleneckUNet (class in src.model.ArtificialModel) ResBottleneckUNetSNN (class in src.model.SpikingModel) ResConv2dBlock (class in src.model.ArtificialBlock) reset_neuron() (src.model.SpikingLayer.NeuronModel method) S save_checkpoint() (src.model.TrainValidTestManager.TrainValidTestManager method) save_coefficients() (src.model.TrainValidTestManager.TrainValidTestManager method) (src.stft.StftManager.StftManager method) save_json_file() (src.model.SpeechEnhancer.SpeechEnhancer method) save_mem (src.model.SpikingLayer.NeuronModel property) (src.model.SpikingModel.SNNBase property) save_transform_info() (src.stft.StftManager.StftManager method) ScaleLayer (class in src.model.SpikingLayer) setup() (src.model.SpeechEnhancer.SpeechEnhancer method) shift() (src.data.TransformManager.TransformManager method) si_sdr_fn() (src.evaluation.EvaluationManager.EvaluationManager static method) si_sdr_loss() (src.model.LossManager.LossManager method) si_snr_fn() (src.evaluation.EvaluationManager.EvaluationManager static method) si_snr_loss() (src.model.LossManager.LossManager method) SigmoidDerivative (class in src.model.SurrogateGradient) SNNBase (class in src.model.SpikingModel) SpeechEnhancer (class in src.model.SpeechEnhancer) spiking_mode (src.model.SurrogateGradient.ATan attribute) (src.model.SurrogateGradient.PiecewiseLinear attribute) (src.model.SurrogateGradient.SigmoidDerivative attribute) (src.model.SurrogateGradient.SuperSpike attribute) SpikingConv2dBlock (class in src.model.SpikingBlock) SpikingFCBlock (class in src.model.SpikingBlock) SpikingResConv2dBlock (class in src.model.SpikingBlock) spk_res_skip_connection() (src.model.SpikingBlock.SpikingResConv2dBlock method) spk_skip_connection() (src.model.SpikingModel.ResBottleneckUNetSNN method) (src.model.SpikingModel.UNetSNN method) standardize() (src.data.TransformManager.TransformManager method) Stft (class in src.stft.StftManager) stft_fn() (src.stft.StftManager.Stft method) StftManager (class in src.stft.StftManager) stoi_fn() (src.evaluation.EvaluationManager.EvaluationManager static method) (src.model.LossManager.LossManager method) stoi_loss() (src.model.LossManager.LossManager method) SuperSpike (class in src.model.SurrogateGradient) surrogate_scale (src.model.SurrogateGradient.ATan attribute) (src.model.SurrogateGradient.PiecewiseLinear attribute) (src.model.SurrogateGradient.SigmoidDerivative attribute) (src.model.SurrogateGradient.SuperSpike attribute) T test_model() (src.model.TrainValidTestManager.TrainValidTestManager method) time_mse_loss() (src.model.LossManager.LossManager method) train_model() (src.model.TrainValidTestManager.TrainValidTestManager method) training (src.model.ArtificialBlock.Conv2dBlock attribute) (src.model.ArtificialBlock.ResConv2dBlock attribute) (src.model.ArtificialModel.ANNBase attribute) (src.model.ArtificialModel.CNN attribute) (src.model.ArtificialModel.ResBottleneckUNet attribute) (src.model.ArtificialModel.UNet attribute) (src.model.LossManager.LossManager attribute) (src.model.SpikingBlock.ReadoutConv2dBlock attribute) (src.model.SpikingBlock.ReadoutFCBlock attribute) (src.model.SpikingBlock.SpikingConv2dBlock attribute) (src.model.SpikingBlock.SpikingFCBlock attribute) (src.model.SpikingBlock.SpikingResConv2dBlock attribute) (src.model.SpikingLayer.LI1d attribute) (src.model.SpikingLayer.LI2d attribute) (src.model.SpikingLayer.LIF1d attribute) (src.model.SpikingLayer.LIF2d attribute) (src.model.SpikingLayer.Linear1d attribute) (src.model.SpikingLayer.MPNeuronModel attribute) (src.model.SpikingLayer.NeuronModel attribute) (src.model.SpikingLayer.Recurrent1d attribute) (src.model.SpikingLayer.Recurrent2d attribute) (src.model.SpikingLayer.ScaleLayer attribute) (src.model.SpikingLayer.Upsampling2d attribute) (src.model.SpikingModel.CSNN attribute) (src.model.SpikingModel.FCSNN attribute) (src.model.SpikingModel.ResBottleneckUNetSNN attribute) (src.model.SpikingModel.SNNBase attribute) (src.model.SpikingModel.UNetSNN attribute) TrainValidTestManager (class in src.model.TrainValidTestManager) TransformManager (class in src.data.TransformManager) U UNet (class in src.model.ArtificialModel) UNetSNN (class in src.model.SpikingModel) update_nb_steps() (src.model.SpikingModel.SNNBase method) update_transform_info() (src.stft.StftManager.StftManager method) Upsampling2d (class in src.model.SpikingLayer) V validate_model() (src.model.TrainValidTestManager.TrainValidTestManager method) VisualizationManager (class in src.visualization.VisualizationManager) visualize_data() (src.model.SpeechEnhancer.SpeechEnhancer method) X x_skip_connection() (src.model.ArtificialModel.ResBottleneckUNet method) (src.model.ArtificialModel.UNet method)