Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/ZhangXInFD/SpeechTokenizer
/ functions
Functions
140 in github.com/ZhangXInFD/SpeechTokenizer
⨍
Functions
140
◇
Types & classes
28
Method
delta
Return the current range width.
speechtokenizer/quantization/ac.py:107
Method
delta
(self)
speechtokenizer/quantization/ac.py:198
Method
device
(self)
speechtokenizer/trainer/trainer.py:270
Method
encode
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
Method
encode
(self, x)
speechtokenizer/quantization/core_vq.py:280
Method
encode
(self, x: torch.Tensor, n_q: tp.Optional[int] = None, st: tp.Optional[int]= None)
speechtokenizer/quantization/core_vq.py:347
Method
forward
(self, x)
speechtokenizer/discriminators.py:36
Method
forward
(self, y, y_hat)
speechtokenizer/discriminators.py:69
Method
forward
(self, x)
speechtokenizer/discriminators.py:100
Method
forward
(self, y, y_hat)
speechtokenizer/discriminators.py:126
Method
forward
(self, x: torch.Tensor)
speechtokenizer/discriminators.py:212
Method
forward
(self, y: torch.Tensor, y_hat: torch.Tensor)
speechtokenizer/discriminators.py:253
Method
forward
Parameters ---------- x : torch.tensor Input wavs. Shape: (batch, channels, timesteps). n_q : in
speechtokenizer/model.py:80
Method
forward
(self, x)
speechtokenizer/modules/seanet.py:36
Method
forward
(self, x)
speechtokenizer/modules/seanet.py:80
Method
forward
(self, x)
speechtokenizer/modules/seanet.py:163
Method
forward
(self, z)
speechtokenizer/modules/seanet.py:256
Method
forward
(self, x)
speechtokenizer/modules/norm.py:24
Method
forward
(self, x)
speechtokenizer/modules/conv.py:119
Method
forward
(self, x)
speechtokenizer/modules/conv.py:136
Method
forward
(self, x)
speechtokenizer/modules/conv.py:153
Method
forward
(self, x)
speechtokenizer/modules/conv.py:169
Method
forward
(self, x)
speechtokenizer/modules/conv.py:195
Method
forward
(self, x)
speechtokenizer/modules/conv.py:230
Method
forward
(self, x)
speechtokenizer/modules/lstm.py:23
Method
forward
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
Method
forward
(self, x)
speechtokenizer/quantization/core_vq.py:204
Method
forward
(self, x)
speechtokenizer/quantization/core_vq.py:292
Method
forward
(self, x, n_q: tp.Optional[int] = None, layers: tp.Optional[list] = None)
speechtokenizer/quantization/core_vq.py:324
Method
forward_feature
Parameters ---------- x : torch.tensor Input wavs. Shape should be (batch, channels, timesteps). layers
speechtokenizer/model.py:113
Function
init_weights
(m, mean=0.0, std=0.01)
speechtokenizer/discriminators.py:16
Method
is_distributed
(self)
speechtokenizer/trainer/trainer.py:274
Method
is_local_main
(self)
speechtokenizer/trainer/trainer.py:282
Method
is_main
(self)
speechtokenizer/trainer/trainer.py:278
Function
pad_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
Function
rank
()
speechtokenizer/quantization/distrib.py:14
Function
sync_buffer
Sync grad for buffers. If average is False, broadcast instead of averaging.
speechtokenizer/quantization/distrib.py:73
Function
sync_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
Function
t_axis_distill_loss
(feature, target_feature, lambda_sim=1)
speechtokenizer/trainer/loss.py:91
Function
uniform_init
(*shape: int)
speechtokenizer/quantization/core_vq.py:55
← previous
101–140 of 140, ranked by callers