MCPcopy Create free account

hub / github.com/ZhangXInFD/SpeechTokenizer / functions

Functions140 in github.com/ZhangXInFD/SpeechTokenizer

Methoddelta
Return the current range width.
speechtokenizer/quantization/ac.py:107
Methoddelta
(self)
speechtokenizer/quantization/ac.py:198
Methoddevice
(self)
speechtokenizer/trainer/trainer.py:270
Methodencode
Encode a given input tensor with the specified sample rate at the given bandwidth. The RVQ encode method sets the appropriate number of quanti
speechtokenizer/quantization/vq.py:87
Methodencode
(self, x)
speechtokenizer/quantization/core_vq.py:280
Methodencode
(self, x: torch.Tensor, n_q: tp.Optional[int] = None, st: tp.Optional[int]= None)
speechtokenizer/quantization/core_vq.py:347
Methodforward
(self, x)
speechtokenizer/discriminators.py:36
Methodforward
(self, y, y_hat)
speechtokenizer/discriminators.py:69
Methodforward
(self, x)
speechtokenizer/discriminators.py:100
Methodforward
(self, y, y_hat)
speechtokenizer/discriminators.py:126
Methodforward
(self, x: torch.Tensor)
speechtokenizer/discriminators.py:212
Methodforward
(self, y: torch.Tensor, y_hat: torch.Tensor)
speechtokenizer/discriminators.py:253
Methodforward
Parameters ---------- x : torch.tensor Input wavs. Shape: (batch, channels, timesteps). n_q : in
speechtokenizer/model.py:80
Methodforward
(self, x)
speechtokenizer/modules/seanet.py:36
Methodforward
(self, x)
speechtokenizer/modules/seanet.py:80
Methodforward
(self, x)
speechtokenizer/modules/seanet.py:163
Methodforward
(self, z)
speechtokenizer/modules/seanet.py:256
Methodforward
(self, x)
speechtokenizer/modules/norm.py:24
Methodforward
(self, x)
speechtokenizer/modules/conv.py:119
Methodforward
(self, x)
speechtokenizer/modules/conv.py:136
Methodforward
(self, x)
speechtokenizer/modules/conv.py:153
Methodforward
(self, x)
speechtokenizer/modules/conv.py:169
Methodforward
(self, x)
speechtokenizer/modules/conv.py:195
Methodforward
(self, x)
speechtokenizer/modules/conv.py:230
Methodforward
(self, x)
speechtokenizer/modules/lstm.py:23
Methodforward
Residual vector quantization on the given input tensor. Args: x (torch.Tensor): Input tensor. n_q (int): Number of qua
speechtokenizer/quantization/vq.py:69
Methodforward
(self, x)
speechtokenizer/quantization/core_vq.py:204
Methodforward
(self, x)
speechtokenizer/quantization/core_vq.py:292
Methodforward
(self, x, n_q: tp.Optional[int] = None, layers: tp.Optional[list] = None)
speechtokenizer/quantization/core_vq.py:324
Methodforward_feature
Parameters ---------- x : torch.tensor Input wavs. Shape should be (batch, channels, timesteps). layers
speechtokenizer/model.py:113
Functioninit_weights
(m, mean=0.0, std=0.01)
speechtokenizer/discriminators.py:16
Methodis_distributed
(self)
speechtokenizer/trainer/trainer.py:274
Methodis_local_main
(self)
speechtokenizer/trainer/trainer.py:282
Methodis_main
(self)
speechtokenizer/trainer/trainer.py:278
Functionpad_for_conv1d
Pad for a convolution to make sure that the last window is full. Extra padding is added at the end. This is required to ensure that we can rebuild
speechtokenizer/modules/conv.py:64
Functionrank
()
speechtokenizer/quantization/distrib.py:14
Functionsync_buffer
Sync grad for buffers. If average is False, broadcast instead of averaging.
speechtokenizer/quantization/distrib.py:73
Functionsync_grad
Simpler alternative to DistributedDataParallel, that doesn't rely on any black magic. For simple models it can also be as fast. Just call
speechtokenizer/quantization/distrib.py:95
Functiont_axis_distill_loss
(feature, target_feature, lambda_sim=1)
speechtokenizer/trainer/loss.py:91
Functionuniform_init
(*shape: int)
speechtokenizer/quantization/core_vq.py:55
← previous101–140 of 140, ranked by callers