Index

A

add_white_noise() (transforms.AddWhiteNoise method) AddWhiteNoise (class in transforms)

Audio_Dataset (class in dataset) AudioVAE (class in models)

B

before_update() (plugins.ekfac_plugin.EKFAC_Plugin method) (plugins.ekfac_plugin.KFAC_Plugin method)

C

CachedAudio (class in dataset) Circularize (class in models) compute_new_kernel_size() (in module matchbox.ConvBlock)

ConvASRDecoderClassification (class in matchbox.ConvASRDecoder) ConvASREncoder (class in matchbox.ConvASREncoder) ConvBlock (class in matchbox.ConvBlock)

D

dataset module

decoder() (models.AudioVAE method)

E

EKFAC (class in plugins.natgrad.ekfac) EKFAC_Plugin (class in plugins.ekfac_plugin)

EncDecBaseModel (class in models) encoder() (models.AudioVAE method) eval_epoch() (templates.river.RiverTemplate method)

F

forward() (matchbox.ConvASRDecoder.ConvASRDecoderClassification method) (matchbox.ConvASREncoder.ConvASREncoder method) (matchbox.ConvBlock.ConvBlock method) (matchbox.ConvBlock.GroupShuffle method) (matchbox.MaskedConv1d.MaskedConv1d method) (models.AudioVAE method) (models.Circularize method) (models.EncDecBaseModel method) (models.M5 method) (models.Pool method) (models.Scattering_Classifier method) (templates.river.RiverTemplate method) (transforms.AddWhiteNoise method) (transforms.MfccTransform method) (transforms.Scattering method) (transforms.SpecAugment method)

G

generate() (models.AudioVAE method) get_metadata() (dataset.MLcommonsData method) get_same_padding() (in module matchbox.ConvBlock)

get_seq_len() (matchbox.MaskedConv1d.MaskedConv1d method) grad_wrt_kernel() (in module plugins.natgrad.ekfac) GroupShuffle (class in matchbox.ConvBlock)

I

infer module

init_weights() (in module matchbox.ConvBlock)

K

KFAC (class in plugins.natgrad.kfac)

KFAC_Plugin (class in plugins.ekfac_plugin)

M

M5 (class in models) main module make_optimizer() (templates.river.RiverTemplate method) MaskedConv1d (class in matchbox.MaskedConv1d) matchbox module matchbox.ConvASRDecoder module matchbox.ConvASREncoder module matchbox.ConvBlock module matchbox.MaskedConv1d module mfcc_transform() (transforms.MfccTransform method) MfccTransform (class in transforms) MLCommons() (dataset.Audio_Dataset method)

MLcommonsData (class in dataset) models module module dataset infer main matchbox matchbox.ConvASRDecoder matchbox.ConvASREncoder matchbox.ConvBlock matchbox.MaskedConv1d models plugins.ekfac_plugin plugins.natgrad plugins.natgrad.ekfac plugins.natgrad.kfac templates.river transforms

P

plugins.ekfac_plugin module plugins.natgrad module plugins.natgrad.ekfac module

plugins.natgrad.kfac module Pool (class in models) predict() (in module infer) preprocess_and_save_dataset() (in module dataset)

R

record() (in module infer) reparameterize() (models.AudioVAE method)

RiverTemplate (class in templates.river) run() (in module main)

S

Scattering (class in transforms) Scattering_Classifier (class in models) spec_aug() (transforms.SpecAugment method) SpecAugment (class in transforms)

speech_commands_collate() (in module dataset) SpeechCommands() (dataset.Audio_Dataset method) SpeechCommandsData (class in dataset) step() (plugins.natgrad.ekfac.EKFAC method) (plugins.natgrad.kfac.KFAC method)

T

tds_normal_() (in module matchbox.ConvBlock) tds_uniform_() (in module matchbox.ConvBlock) templates.river module

tensor_to_dict() (templates.river.RiverTemplate method) training_epoch() (templates.river.RiverTemplate method) transforms module

U

update_masked_length() (matchbox.MaskedConv1d.MaskedConv1d method)