Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/dangf15/THLNet
/ functions
Functions
68 in github.com/dangf15/THLNet
⨍
Functions
68
◇
Types & classes
24
↓ 5 callers
Method
__init__
(self, rnn_type, input_size, hidden_size, output_size, freq_dim, down_sample, dropout=0, num
nets/dpcrn_block.py:147
↓ 4 callers
Method
__init__
(self, complex_axis=1)
complexnn.py:14
↓ 4 callers
Method
__init__
(self, width=64, input_channel_rate=1, output_channel=2)
nets/dfnet_block.py:223
↓ 2 callers
Method
__init__
(self, start_epoch=None, start_iter=None, train_loss=None, eval_loss=None, best_val_loss=float("inf"),
utils.py:13
↓ 2 callers
Method
__init__
(self)
nets/model.py:209
↓ 2 callers
Method
forward
(self, input)
nets/dpcrn_block.py:175
↓ 2 callers
Function
init_kernels
(win_len, win_inc, fft_len, win_type=None, invers=False)
conv_stft.py:8
↓ 1 callers
Method
__init__
(self, win_len, win_inc, fft_len=None, win_type='hamming', feature_type='real', fix=True)
conv_stft.py:30
↓ 1 callers
Function
_get_activation_fn
(activation)
nets/dfnet_block.py:96
↓ 1 callers
Function
complex_cat
(inputs, axis)
complexnn.py:67
↓ 1 callers
Method
reset_parameters
(self)
complexnn.py:269
↓ 1 callers
Method
reset_running_stats
(self)
complexnn.py:260
↓ 1 callers
Function
show_model
(nnet)
show.py:21
↓ 1 callers
Function
show_params
(nnet)
show.py:6
↓ 1 callers
Function
test_ifft1
()
conv_stft.py:134
Method
__init__
(self, num_features, affine=True, eps=1e-5,
utils.py:60
Method
__init__
(self, num_features, affine=True, eps=1e-5,
utils.py:87
Method
__init__
(self, win_len, win_inc, fft_len=None, win_type='hamming', feature_type='real', fix=True)
conv_stft.py:64
Method
__init__
(self, input_size, hidden_size, projection_dim=None, bidirectional=False, batch_first=False)
complexnn.py:28
Method
__init__
in_channels: real+imag out_channels: real+imag kernel_size : input [B,C,D,T] kernel size in [D,T]
complexnn.py:81
Method
__init__
in_channels: real+imag out_channels: real+imag
complexnn.py:151
Method
__init__
(self, num_features, eps=1e-5, momentum=0.1, affine=True, track_running_stats=True, complex_axis=
complexnn.py:220
Method
__init__
(self, d_model, nhead, bidirectional=True, dropout=0, activation="relu")
nets/dfnet_block.py:37
Method
__init__
(self, input_size, output_size, dropout=0, num_layers=1)
nets/dfnet_block.py:120
Method
__init__
(self, in_channels, out_channels, kernel_size, r=1)
nets/dfnet_block.py:176
Method
__init__
(self, depth=5, in_channels=64)
nets/dfnet_block.py:194
Method
__init__
(self, in_channels, out_channels, kernel_size, r=1)
nets/dpcrn_block.py:12
Method
__init__
(self, dim, elementwise_affine=True)
nets/dpcrn_block.py:35
Method
__init__
(self, dim1, dim2, eps=1e-8, elementwise_affine=True)
nets/dpcrn_block.py:68
Method
__init__
(self, rnn_type, input_size, hidden_size, dropout=0, bidirectional=False)
nets/dpcrn_block.py:110
Method
__init__
rnn_layers: the number of lstm layers in the crn, rnn_units: for clstm, rnn_units = real+imag
nets/dpcrn_block.py:206
Method
__init__
( self, width=48, win_len=512,
nets/model.py:15
Method
__init__
( self, width=48, win_len=512,
nets/model.py:110
Method
__setstate__
(self, state)
nets/dfnet_block.py:63
Method
_check_input_dim
(self, xr, xi)
complexnn.py:278
Function
_get_clones
(module, N)
nets/dfnet_block.py:92
Function
checkcausal_net
The whole pipeline give a latency of aec_shift + net win = 48ms
utils.py:112
Method
extra_repr
(self)
complexnn.py:395
Method
flatten_parameters
(self)
complexnn.py:63
Method
forward
(self, inpt)
utils.py:77
Method
forward
(self, inpt)
utils.py:103
Method
forward
(self, inputs)
conv_stft.py:46
Method
forward
inputs : [B, N+2, T] (complex spec) or [B, N//2+1, T] (mags) phase: [B, N//2+1, T] (if not none)
conv_stft.py:82
Method
forward
(self, inputs)
complexnn.py:21
Method
forward
(self, inputs)
complexnn.py:46
Method
forward
(self,inputs)
complexnn.py:121
Method
forward
(self,inputs)
complexnn.py:187
Method
forward
(self, inputs)
complexnn.py:282
Method
forward
r"""Pass the input through the encoder layer. Args: src: the sequnce to the encoder layer (required). src_mask: the
nets/dfnet_block.py:68
Method
forward
(self, input)
nets/dfnet_block.py:149
Method
forward
(self, x)
nets/dfnet_block.py:184
Method
forward
(self, x)
nets/dfnet_block.py:212
Method
forward
(self, x)
nets/dfnet_block.py:257
Method
forward
(self, x)
nets/dpcrn_block.py:20
Method
forward
(self, x)
nets/dpcrn_block.py:39
Method
forward
(self, x)
nets/dpcrn_block.py:82
Method
forward
(self, input)
nets/dpcrn_block.py:124
Method
forward
(self, inputs, lens=None)
nets/dpcrn_block.py:326
Method
forward
(self, inputs, lens=None)
nets/model.py:59
Method
forward
(self, x1, x2,lens=None)
nets/model.py:145
Method
forward
(self, x)
nets/model.py:219
Function
get_casual_padding1d
()
complexnn.py:6
Function
get_casual_padding2d
()
complexnn.py:9
Method
get_params
(self, weight_decay=0.0)
nets/model.py:230
Method
load
(self, filename)
utils.py:39
Method
save
(self, is_best, filename, best_model)
utils.py:30
Function
test_fft
()
conv_stft.py:103
Function
test_ifft2
()
conv_stft.py:153