Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/HA-Transformer/MAT
/ functions
Functions
2,038 in github.com/HA-Transformer/MAT
⨍
Functions
2,038
◇
Types & classes
299
↓ 5 callers
Method
_map_index
(self, key, index)
fairseq/data/round_robin_zip_datasets.py:42
↓ 5 callers
Method
assertTensorEqual
(self, t1, t2)
tests/speech_recognition/test_collaters.py:54
↓ 5 callers
Function
base_architecture
(args)
fairseq/models/fconv.py:603
↓ 5 callers
Function
base_architecture
(args)
fairseq/models/transformer.py:526
↓ 5 callers
Function
binarize
(args, filename, vocab, output_prefix, lang, offset, end, append_eos=True)
preprocess.py:234
↓ 5 callers
Method
detokenize
(self, sentence: str)
fairseq/hub_utils.py:160
↓ 5 callers
Method
encode
(self, line)
examples/roberta/multiprocessing_bpe_encoder.py:98
↓ 5 callers
Function
get_parser
(desc, default_task='translation')
fairseq/options.py:158
↓ 5 callers
Method
load_dictionary
Load the dictionary from the filename Args: filename (str): the filename
examples/roberta/wsc/wsc_task.py:60
↓ 5 callers
Function
make_dataset
(path, impl, fix_lua_indexing=False, dictionary=None)
fairseq/data/indexed_dataset.py:51
↓ 5 callers
Function
merge
(key, is_list=False)
fairseq/data/monolingual_dataset.py:16
↓ 5 callers
Method
save_checkpoint
Save all training state in a checkpoint file.
fairseq/trainer.py:124
↓ 5 callers
Method
setUpClass
(cls)
tests/speech_recognition/asr_test_base.py:391
↓ 5 callers
Method
setUpEncoder
(self, encoder)
tests/speech_recognition/asr_test_base.py:396
↓ 5 callers
Method
setUpInput
(self, input=None)
tests/speech_recognition/asr_test_base.py:403
↓ 5 callers
Method
step
Take a single search step. Args: step: the current search step, starting at 0 lprobs: (bsz x input_beam_size x vocab_
fairseq/search.py:28
↓ 5 callers
Function
train_path
(lang)
preprocess.py:33
↓ 5 callers
Function
train_path
(lang)
fairseq_cli/preprocess.py:33
↓ 5 callers
Method
zero_grad
(self)
fairseq/trainer.py:459
↓ 4 callers
Function
Embedding
(num_embeddings, embedding_dim, padding_idx)
fairseq/models/lstm.py:471
↓ 4 callers
Function
Linear
(in_features, out_features, bias=True)
fairseq/modules/transformer_layer.py:274
↓ 4 callers
Function
Linear
(in_features, out_features, bias=True)
fairseq/models/transformer_mb/transformer_mb_layer.py:379
↓ 4 callers
Method
__init__
(self, iterable, epoch=None, prefix=None)
fairseq/progress_bar.py:69
↓ 4 callers
Method
__init__
(self, tgt_dict)
fairseq/search.py:13
↓ 4 callers
Method
__init__
(self, encoder, decoder)
fairseq/models/lightconv.py:50
↓ 4 callers
Method
__init__
(self, encoder, decoder)
tests/utils.py:156
↓ 4 callers
Method
_format_stats
(self, stats)
fairseq/progress_bar.py:107
↓ 4 callers
Method
_get_input_buffer
(self, incremental_state)
fairseq/modules/dynamic_convolution.py:208
↓ 4 callers
Method
_in_proj
(self, input, start=0, end=None)
fairseq/modules/multihead_attention.py:315
↓ 4 callers
Function
_lang_token_index
Return language token index.
fairseq/tasks/multilingual_translation.py:29
↓ 4 callers
Method
_map_index_to_dataset
Different underlying datasets have different lengths. In order to ensure we are not accessing an index outside the range of the curre
fairseq/data/multi_corpus_sampled_dataset.py:71
↓ 4 callers
Function
_pair
(v)
fairseq/modules/vggblock.py:15
↓ 4 callers
Method
_prepare_sample
(self, sample)
fairseq/trainer.py:495
↓ 4 callers
Method
_transpose_if_training
(self, x, incremental_state)
fairseq/models/fconv.py:538
↓ 4 callers
Function
add_dataset_args
(parser, train=False, gen=False)
fairseq/options.py:258
↓ 4 callers
Method
add_item
(self, tensor)
fairseq/data/indexed_dataset.py:304
↓ 4 callers
Function
all_reduce
(tensor, group=None)
fairseq/distributed_utils.py:123
↓ 4 callers
Method
assertAlmostEqual
(self, t1, t2)
tests/test_convtbc.py:42
↓ 4 callers
Method
assertAlmostEqual
(self, t1, t2)
tests/test_utils.py:77
↓ 4 callers
Method
assertAlmostEqual
(self, t1, t2)
tests/test_character_token_embedder.py:40
↓ 4 callers
Method
assertTensorEqual
(self, t1, t2)
tests/test_noising.py:520
↓ 4 callers
Function
base_architecture
(args)
fairseq/models/lightconv.py:666
↓ 4 callers
Function
base_architecture
(args)
fairseq/models/transformer_mb/transformer_mb.py:598
↓ 4 callers
Method
binarize
(filename, dict, consumer, tokenize=tokenize_line, append_eos=True, reverse_order=False, offs
fairseq/binarizer.py:25
↓ 4 callers
Function
bleu_addngram
fairseq/clib/libbleu/libbleu.cpp:69
↓ 4 callers
Method
build_optimizer
Args: args (argparse.Namespace): fairseq args params (iterable): iterable of parameters to optimize
fairseq/optim/fp16_optimizer.py:85
↓ 4 callers
Function
dataset_dest_file
(args, output_prefix, lang, extension)
preprocess.py:255
↓ 4 callers
Function
dataset_dest_file
(args, output_prefix, lang, extension)
fairseq_cli/preprocess.py:255
↓ 4 callers
Function
dictolist
(d)
examples/translation_moe/score.py:47
↓ 4 callers
Method
find_offsets
(filename, num_chunks)
fairseq/binarizer.py:56
↓ 4 callers
Function
get_encoder
(encoder_json_path, vocab_bpe_path)
fairseq/data/encoders/gpt2_bpe_utils.py:118
↓ 4 callers
Method
get_lr
Get the current learning rate.
fairseq/trainer.py:472
↓ 4 callers
Method
get_model
Get the (non-wrapped) model instance.
fairseq/trainer.py:476
↓ 4 callers
Method
get_normalized_probs
(self, net_output, log_probs, _)
tests/utils.py:225
↓ 4 callers
Method
get_original_text
(self, i)
fairseq/data/indexed_dataset.py:264
↓ 4 callers
Function
get_res_file
(file_prefix)
examples/speech_recognition/infer.py:106
↓ 4 callers
Method
get_train_iterator
Return an EpochBatchIterator over the training set for a given epoch.
fairseq/trainer.py:198
↓ 4 callers
Method
in_proj_k
(self, key)
fairseq/modules/multihead_attention.py:295
↓ 4 callers
Method
in_proj_q
(self, query)
fairseq/modules/multihead_attention.py:286
↓ 4 callers
Method
in_proj_v
(self, value)
fairseq/modules/multihead_attention.py:305
↓ 4 callers
Method
lr_step
Adjust the learning rate based on the validation loss.
fairseq/trainer.py:462
↓ 4 callers
Method
maybe_layer_norm
(self, layer_norm, x, before=False, after=False)
fairseq/modules/transformer_layer.py:109
↓ 4 callers
Method
maybe_layer_norm
(self, i, x, before=False, after=False)
fairseq/models/lightconv.py:505
↓ 4 callers
Method
maybe_layer_norm
(self, layer_norm, x, before=False, after=False)
fairseq/models/transformer_mb/transformer_mb_layer.py:167
↓ 4 callers
Method
model
(self)
fairseq/trainer.py:79
↓ 4 callers
Method
noising
(self, x, lengths, noising_prob=0.0)
fairseq/data/noising.py:34
↓ 4 callers
Method
noising
(self, x, lengths, dropout_prob=None, blank_idx=None)
fairseq/data/noising.py:77
↓ 4 callers
Function
preprocess_lm_data
(data_dir)
tests/test_binaries.py:528
↓ 4 callers
Function
readlines
(fd)
score.py:47
↓ 4 callers
Function
readlines
(fd)
fairseq_cli/score.py:47
↓ 4 callers
Method
register_classification_head
Register a classification head.
fairseq/models/roberta/model.py:105
↓ 4 callers
Method
result_string
(self, order=4)
fairseq/bleu.py:339
↓ 4 callers
Method
self_attn
(self, query, key, value, **kwargs)
fairseq/models/transformer_mb/transformer_mb_layer.py:101
↓ 4 callers
Method
setUp
(self)
tests/speech_recognition/test_vggtransformer.py:126
↓ 4 callers
Function
strip_pronoun
(x)
examples/roberta/wsc/wsc_utils.py:115
↓ 4 callers
Method
train_step
Do forward, backward and parameter update.
fairseq/trainer.py:219
↓ 3 callers
Function
Embedding
(num_embeddings, embedding_dim, padding_idx)
fairseq/models/transformer.py:510
↓ 3 callers
Function
GatedLinear
Weight-normalized Linear layer (input: B x T x C) with interspersed GLU units
fairseq/modules/downsampled_multihead_attention.py:248
↓ 3 callers
Method
__init__
(self, index)
fairseq/modules/downsampled_multihead_attention.py:232
↓ 3 callers
Method
__init__
(self, dictionary, bpe_cont_marker="@@", bpe_end_marker=None)
fairseq/data/noising.py:14
↓ 3 callers
Method
__init__
(self, encoder, decoder)
fairseq/models/fconv.py:52
↓ 3 callers
Method
__init__
( self, dictionary, embed_dim=512, max_positions=1024, convolutions=((512, 3),) * 20, dropout=
fairseq/models/fconv_self_att.py:151
↓ 3 callers
Method
__init__
(self, encoder, decoder)
fairseq/models/lstm.py:23
↓ 3 callers
Method
__init__
(self, args, encoder)
fairseq/models/roberta/model.py:41
↓ 3 callers
Method
_backtranslation_dataset_helper
( self, remove_eos_from_input_src, remove_eos_from_output_src, )
tests/test_backtranslation_dataset.py:32
↓ 3 callers
Method
_build_optimizer
(self)
fairseq/trainer.py:101
↓ 3 callers
Method
_convert_src_tokens_to_tensor
( self, vocab: Dictionary, src_tokens: List[List[str]], append_eos: bool )
tests/test_noising.py:122
↓ 3 callers
Function
_get_bt_dataset_key
(lang_pair)
fairseq/tasks/semisupervised_translation.py:26
↓ 3 callers
Function
_get_denoising_dataset_key
(lang_pair)
fairseq/tasks/semisupervised_translation.py:30
↓ 3 callers
Method
_init_buffers
(self, t)
fairseq/search.py:22
↓ 3 callers
Method
_sync_fp16_grads_to_fp32
(self, multiply_grads=1.)
fairseq/optim/fp16_optimizer.py:148
↓ 3 callers
Method
_test_pretrained_masked_lm_for_translation
(self, learned_pos_emb, encoder_only)
tests/test_binaries.py:275
↓ 3 callers
Method
_test_reproducibility
(self, name, extra_flags=None)
tests/test_reproducibility.py:18
↓ 3 callers
Method
_unscale_grads
(self, multiply_grads=1.)
fairseq/optim/fp16_optimizer.py:328
↓ 3 callers
Function
add_common_eval_args
(group)
fairseq/options.py:408
↓ 3 callers
Method
add_file_to_dictionary
(filename, dict, tokenize, num_workers)
fairseq/data/dictionary.py:286
↓ 3 callers
Method
aggregate_logging_outputs
(self, logging_outputs, criterion)
fairseq/tasks/fairseq_task.py:254
↓ 3 callers
Method
all_reduce
(params)
fairseq/legacy_distributed_data_parallel.py:96
↓ 3 callers
Method
assertAlmostEqual
(self, t1, t2)
tests/test_label_smoothing.py:93
↓ 3 callers
Method
assert_eos_at_end
Asserts last token of every sentence in x is EOS
tests/test_noising.py:140
← previous
next →
101–200 of 2,038, ranked by callers