Functions — github.com/XuezheMax/flowseq

CodeHub MCPcopy Create free account

hub / github.com/XuezheMax/flowseq / functions

Functions372 in github.com/XuezheMax/flowseq

⨍ Functions 372◇ Types & classes 56

(self, dataset, batch_size, batch_by_tokens, max_src_length, max_tgt_length, buffer_multiple_size,

flownmt/data/dataloader.py:246

(self, in_features, inverse=False)

flownmt/flows/actnorm.py:12

(self, inverse)

flownmt/flows/flow.py:12

(self, in_features, heads=None, type='A', inverse=False)

flownmt/flows/linear.py:98

(self, features, src_features, hidden_features=None, inverse=False, transform='affine', heads

flownmt/flows/nmt.py:17

(self, features, src_features, hidden_features=None, inverse=False, transform='affine', coupl

flownmt/flows/nmt.py:77

(self, features, src_features, hidden_features=None, inverse=False, transform='affine', coupl

flownmt/flows/nmt.py:249

(self, num_steps, features, src_features, hidden_features=None, inverse=False, prior=False, factor=2,

flownmt/flows/nmt.py:337

(self, src_features, features, hidden_features=None, inverse=False, split_dim=2, split_type='continuous', orde

flownmt/flows/couplings/coupling.py:14

(self, rnn_mode, src_features, in_features, out_features, hidden_features, dropout=0.0)

flownmt/flows/couplings/blocks.py:52

(self, src_features, in_features, out_features, hidden_features, heads, dropout=0.0, pos_enc=

flownmt/flows/couplings/blocks.py:89

(self, flow: Flow, device_ids=None, output_device=None, dim=0)

flownmt/flows/parallel/data_parallel.py:16

(self, error_queue)

experiments/distributed.py:107

flownmt/data/dataloader.py:440

flownmt/data/dataloader.py:466

Method__setstate__

flownmt/optim/adamw.py:47

flownmt/data/dataloader.py:412

Args: input: Tensor input tensor [batch, N1, N2, ..., Nl, in_features] mask: Tensor

flownmt/flows/actnorm.py:48

Args: input: Tensor input tensor [batch, N1, N2, ..., Nl, in_features] mask: Tensor

flownmt/flows/linear.py:51

Args: input: Tensor input tensor [batch, N1, N2, ..., Nl, in_features] mask: Tensor

flownmt/flows/linear.py:153

(self, input: torch.Tensor, tgt_mask: torch.Tensor, src: torch.Tensor, src_mask: torch.Tensor

flownmt/flows/nmt.py:45

(self, input: torch.Tensor, tgt_mask: torch.Tensor, src: torch.Tensor, src_mask: torch.Tensor

flownmt/flows/nmt.py:163

(self, input: torch.Tensor, tgt_mask: torch.Tensor, src: torch.Tensor, src_mask: torch.Tensor

flownmt/flows/nmt.py:290

(self, input: torch.Tensor, tgt_mask: torch.Tensor, src: torch.Tensor, src_mask: torch.Tensor

flownmt/flows/nmt.py:390

Args: input: Tensor input tensor [batch, length, in_features] mask: Tensor mask tenso

flownmt/flows/couplings/coupling.py:126

Methodbatch_size_fn

flownmt/data/dataloader.py:369

(z: torch.Tensor, mask: torch.Tensor, params)

flownmt/flows/couplings/transform.py:13

(z: torch.Tensor, mask: torch.Tensor, params)

flownmt/flows/couplings/transform.py:28

(z: torch.Tensor, mask: torch.Tensor, params)

flownmt/flows/couplings/transform.py:47

(cls, name: str)

flownmt/modules/encoders/encoder.py:47

(cls, name: str)

flownmt/modules/posteriors/posterior.py:93

(cls, name: str)

flownmt/modules/decoders/decoder.py:85

(cls, name: str)

flownmt/modules/priors/length_predictors/predictor.py:56

Args: z: Tensor latent code [batch, length, hidden_size] mask: Tensor mask [batch, l

flownmt/modules/decoders/decoder.py:38

Args: z: Tensor latent code [batch, length, hidden_size] mask: Tensor mask [batch, l

flownmt/modules/decoders/simple.py:36

Args: z: Tensor latent code [batch, length, hidden_size] mask: Tensor mask [batch, l

flownmt/modules/decoders/transformer.py:48

Args: z: Tensor latent code [batch, length, hidden_size] mask: Tensor mask [batch, l

flownmt/modules/decoders/rnn.py:62

flownmt/data/dataloader.py:463

Methoderror_listener

experiments/distributed.py:117

FunctionexponentialMovingAverage

(original, shadow, decay_rate, init=False)

flownmt/utils.py:34

Methodextra_repr

flownmt/nnet/weightnorm.py:81

Methodextra_repr

flownmt/flows/actnorm.py:99

Methodextra_repr

flownmt/flows/linear.py:80

Methodextra_repr

flownmt/flows/linear.py:192

Methodextra_repr

flownmt/flows/couplings/coupling.py:167

(self, src_sents: torch.Tensor, tgt_sents: torch, src_masks: torch.Tensor, tgt_masks: torch.Tensor,

flownmt/flownmt.py:376

(self, src_sents, masks)

flownmt/modules/encoders/transformer.py:31

(self, src_sents, masks=None)

flownmt/modules/encoders/transformer.py:64

(self, src_sents, masks)

flownmt/modules/encoders/rnn.py:34

(self, src_sents, masks=None)

flownmt/modules/encoders/rnn.py:63

Encoding src sentences into src encoding representations. Args: src_sents: Tensor [batch, length] masks: Tens

flownmt/modules/encoders/encoder.py:25

(self, tgt_sents, tgt_masks, src_enc, src_masks)

flownmt/modules/posteriors/shift_rnn.py:44

(self, tgt_sents, tgt_masks, src_enc, src_masks)

flownmt/modules/posteriors/shift_rnn.py:124

(self, tgt_sents, tgt_masks, src_enc, src_masks)

flownmt/modules/posteriors/posterior.py:56

(self, tgt_sents, tgt_masks, src_enc, src_masks)

flownmt/modules/posteriors/transformer.py:35

(self, tgt_sents, tgt_masks, src_enc, src_masks)

flownmt/modules/posteriors/transformer.py:87

(self, tgt_sents, tgt_masks, src_enc, src_masks)

flownmt/modules/posteriors/rnn.py:40

(self, tgt_sents, tgt_masks, src_enc, src_masks)

flownmt/modules/posteriors/rnn.py:97

(self, z, src, src_mask)

flownmt/modules/decoders/simple.py:25

(self, z, mask, src, src_mask)

flownmt/modules/decoders/transformer.py:28

(self, z, mask, src, src_mask)

flownmt/modules/decoders/rnn.py:42

flownmt/modules/priors/length_predictors/diff_discretized_mix_logistic.py:33

flownmt/modules/priors/length_predictors/diff_softmax.py:40

Input is expected to be of size [bsz x seqlen].

flownmt/nnet/positional_encoding.py:41

Args: query: Tensor query tensor [batch, query_length, query_features] key: Tensor k

flownmt/nnet/attention.py:52

Args: query: Tenfor [batch, tgt_len, model_dim] key: Tensor [batch, src_len, model_d

flownmt/nnet/attention.py:133

flownmt/nnet/attention.py:243

(self, input, target)

flownmt/nnet/criterion.py:14

(self, x, mask)

flownmt/nnet/transformer.py:12

(self, x, mask, src, src_mask)

flownmt/nnet/transformer.py:30

flownmt/nnet/weightnorm.py:40

flownmt/nnet/weightnorm.py:77

(self, input: torch.Tensor, tgt_mask: torch.Tensor, src: torch.Tensor, src_mask: torch.Tensor)

flownmt/flows/nmt.py:32

(self, input: torch.Tensor, tgt_mask: torch.Tensor, src: torch.Tensor, src_mask: torch.Tensor)

flownmt/flows/nmt.py:123

(self, input: torch.Tensor, tgt_mask: torch.Tensor, src: torch.Tensor, src_mask: torch.Tensor)

flownmt/flows/nmt.py:269

(self, input: torch.Tensor, tgt_mask: torch.Tensor, src: torch.Tensor, src_mask: torch.Tensor)

flownmt/flows/nmt.py:370

Args: input: Tensor input tensor [batch, length, in_features] mask: Tensor mask tenso

flownmt/flows/couplings/coupling.py:98

Args: x: Tensor input tensor [batch, length, in_features] mask: Tensor x mask tensor

flownmt/flows/couplings/blocks.py:20

(self, x, mask, src, src_mask)

flownmt/flows/couplings/blocks.py:67

(self, x, mask, src, src_mask)

flownmt/flows/couplings/blocks.py:99

Methodfrom_params

(cls, params: Dict)

flownmt/modules/encoders/transformer.py:74

Methodfrom_params

(cls, params: Dict)

flownmt/modules/encoders/rnn.py:73

Methodfrom_params

(cls, params: Dict)

flownmt/modules/encoders/encoder.py:51

Methodfrom_params

(cls, params: Dict)

flownmt/modules/posteriors/shift_rnn.py:147

Methodfrom_params

(cls, params: Dict)

flownmt/modules/posteriors/posterior.py:97

Methodfrom_params

(cls, params: Dict)

flownmt/modules/posteriors/transformer.py:110

Methodfrom_params

(cls, params: Dict)

flownmt/modules/posteriors/rnn.py:120

Methodfrom_params

(cls, params: Dict)

flownmt/modules/decoders/decoder.py:89

Methodfrom_params

(cls, params: Dict)

flownmt/modules/decoders/simple.py:91

Methodfrom_params

(cls, params: Dict)

flownmt/modules/decoders/transformer.py:103

Methodfrom_params

(cls, params: Dict)

flownmt/modules/decoders/rnn.py:117

Methodfrom_params

(cls, params: Dict)

flownmt/modules/priors/prior.py:209

Methodfrom_params

(cls, params: Dict)

flownmt/modules/priors/length_predictors/predictor.py:60

Methodfrom_params

(cls, params: Dict)

flownmt/modules/priors/length_predictors/diff_discretized_mix_logistic.py:96

Methodfrom_params

(cls, params: Dict)

flownmt/modules/priors/length_predictors/diff_softmax.py:118

Methodfrom_params

(cls, params: Dict)

flownmt/flows/actnorm.py:103

Methodfrom_params

(cls, params: Dict)

flownmt/flows/flow.py:112

Methodfrom_params

(cls, params: Dict)

flownmt/flows/linear.py:84

Methodfrom_params

(cls, params: Dict)

flownmt/flows/linear.py:196

Methodfrom_params

(cls, params: Dict)

flownmt/flows/nmt.py:533

← previous next →201–300 of 372, ranked by callers