MCPcopy Create free account

hub / github.com/OFA-Sys/OFA / functions

Functions6,938 in github.com/OFA-Sys/OFA

⨍ Functions 6,938◇ Types & classes 1,161↳ Endpoints 9
↓ 7 callersMethod_seek
(self, offset=0)
data/file_dataset.py:80
↓ 7 callersMethod_test_reproducibility
( self, name, extra_flags=None, delta=0.0001, resume_checkpoint="check
fairseq/tests/test_reproducibility.py:19
↓ 7 callersMethodadd_item
(self, tensor)
fairseq/fairseq/data/indexed_dataset.py:340
↓ 7 callersMethodadd_meter
(self, key, meter, priority)
fairseq/fairseq/logging/meters.py:272
↓ 7 callersMethodalter_dataset_langtok
( self, lang_pair_dataset, src_eos=None, src_lang=None, tgt_eos=None,
fairseq/fairseq/tasks/multilingual_translation.py:178
↓ 7 callersFunctionbase_lm_architecture
(args)
fairseq/fairseq/models/transformer_lm.py:297
↓ 7 callersMethodbuild_code
( self, bos="<s>", pad="<pad>", eos="</s>", unk="<unk>", )
fairseq/fairseq/data/huffman/huffman_coder.py:218
↓ 7 callersMethodbuild_generator
Build a :class:`~fairseq.SequenceGenerator` instance for this task. Args: models (List[~fairseq.models.FairseqMo
fairseq/fairseq/tasks/fairseq_task.py:339
↓ 7 callersFunctioncompute_mask_indices
Computes random mask spans for a given shape Args: shape: the the shape for which to compute masks. should be of size 2
fairseq/fairseq/data/data_utils.py:393
↓ 7 callersFunctioncreate_dummy_roberta_head_data
( data_dir, num_examples=100, maxlen=10, num_classes=2, regression=False )
fairseq/tests/test_binaries.py:1741
↓ 7 callersFunctioncreate_zip
(data_root: Path, zip_path: Path)
fairseq/examples/speech_to_text/data_utils.py:101
↓ 7 callersMethoddecode
(toks)
tasks/nlg_tasks/gigaword.py:236
↓ 7 callersFunctiongen_vocab
( input_path: Path, output_path_prefix: Path, model_type="bpe", vocab_size=1000, special_symbols: Opti
fairseq/examples/speech_to_text/data_utils.py:32
↓ 7 callersMethodgenerate_unchanged_shuffle_map
(self, length)
fairseq/tests/test_noising.py:191
↓ 7 callersFunctionget_active_aggregators
()
fairseq/fairseq/logging/metrics.py:107
↓ 7 callersMethodget_batch_iterator
Get an iterator that yields batches of data from the given dataset. Args: dataset (~fairseq.data.FairseqDataset): datase
fairseq/fairseq/tasks/fairseq_task.py:208
↓ 7 callersMethodget_criterion
Get the (non-wrapped) criterion instance.
trainer.py:1135
↓ 7 callersMethodget_criterion
Get the (non-wrapped) criterion instance.
fairseq/fairseq/trainer.py:1113
↓ 7 callersMethodget_targets
(self, net_output, is_masked=True)
fairseq/fairseq/models/hubert/hubert.py:544
↓ 7 callersFunctionget_zip_manifest
( zip_path: Path, zip_root: Optional[Path] = None, is_audio=False )
fairseq/examples/speech_to_text/data_utils.py:108
↓ 7 callersFunctionindex_file_path
(prefix_path)
fairseq/fairseq/data/indexed_dataset.py:136
↓ 7 callersFunctionlengths_to_mask
(lens)
fairseq/fairseq/data/data_utils.py:541
↓ 7 callersMethodload_dictionary
Load the dictionary from the filename Args: filename (str): the filename
fairseq/examples/roberta/wsc/wsc_task.py:65
↓ 7 callersMethodmake_generation_fast_
(self, need_attn=False, **kwargs)
fairseq/fairseq/models/lstm.py:662
↓ 7 callersFunctionmerge
(key)
data/mm_data/ocr_dataset.py:42
↓ 7 callersMethodset_epoch
(self, epoch)
fairseq/fairseq/data/list_dataset.py:31
↓ 7 callersMethodshuffle
Shuffle dataset samples before batching
fairseq/fairseq/data/audio/data_cfg.py:50
↓ 7 callersFunctionspawn_and_init
(fn, world_size, args=None)
fairseq/tests/distributed/utils.py:12
↓ 7 callersFunctiontrain_roberta_head
(data_dir, arch, num_classes=2, extra_flags=None)
fairseq/tests/test_binaries.py:1813
↓ 7 callersMethodwrite
(self, sizes, pointers)
fairseq/fairseq/data/huffman/huffman_mmap_indexed_dataset.py:43
↓ 7 callersMethodzero_grad
(self)
fairseq/fairseq/trainer.py:1079
↓ 6 callersFunctionLinear
(in_features, out_features, bias=True)
fairseq/fairseq/model_parallel/models/pipeline_parallel_transformer/layers.py:595
↓ 6 callersFunctionLinear
Linear layer (input: N x T x C)
fairseq/fairseq/models/lstm.py:689
↓ 6 callersMethod__init__
(self, embed_dim: int, # vision image_resolution: int,
models/clip/model.py:242
↓ 6 callersMethod_build_dataset
(self, data, **kwargs)
fairseq/tests/test_token_block_dataset.py:14
↓ 6 callersFunction_create_dummy_data
(filename)
fairseq/tests/utils.py:163
↓ 6 callersMethod_set_input_buffer
( self, incremental_state: Dict[str, Dict[str, Optional[Tensor]]], buffer: Dict[str, O
fairseq/fairseq/model_parallel/modules/multihead_attention.py:344
↓ 6 callersFunction_skip
Getting sliced (dim=0) tensor by mask. Supporting tensor and list/dict of tensors.
fairseq/fairseq/models/nat/levenshtein_utils.py:231
↓ 6 callersFunction_skip_encoder_out
(encoder, encoder_out, mask)
fairseq/fairseq/models/nat/levenshtein_utils.py:256
↓ 6 callersMethodattention
(self, x: torch.Tensor)
models/ofa/vit.py:31
↓ 6 callersFunctionbase_architecture
(args)
fairseq/fairseq/models/roberta/model.py:524
↓ 6 callersFunctionbase_architecture
(args)
fairseq/examples/speech_recognition/models/vggtransformer.py:894
↓ 6 callersFunctionbroadcast
(tensor, src, group)
fairseq/fairseq/distributed/utils.py:527
↓ 6 callersMethodbuild_fp32_params
Store a copy of the EMA params in fp32. If state dict is passed, the EMA params is copied from the provided state dict. Other
fairseq/fairseq/models/ema/ema.py:96
↓ 6 callersMethodbuild_progress_bar
( self, epoch: Optional[int] = None, prefix: Optional[str] = None, default_log
fairseq/examples/speech_recognition/new/infer.py:233
↓ 6 callersFunctioncheck_type
(module, expected_type)
fairseq/fairseq/models/fairseq_model.py:30
↓ 6 callersFunctionconvert_waveform
convert a waveform: - to a target sample rate - from multi-channel to mono channel - volume normalization Args:
fairseq/fairseq/data/audio/audio_utils.py:19
↓ 6 callersMethodcost
(self, ref, hyp, code)
fairseq/examples/speech_recognition/utils/wer_utils.py:80
↓ 6 callersMethoddecode
(self, emissions)
fairseq/examples/speech_recognition/w2l_decoder.py:219
↓ 6 callersMethoddetokenize
(self, sentence: str)
fairseq/fairseq/hub_utils.py:238
↓ 6 callersFunctiondict_path
(lang)
fairseq/fairseq_cli/preprocess.py:60
↓ 6 callersFunctiondownsample2
Downsampling the input by 2 using sinc interpolation. Smith, Julius, and Phil Gossett. "A flexible sampling-rate conversion method." ICAS
fairseq/examples/speech_synthesis/preprocessing/denoiser/resample.py:63
↓ 6 callersFunctionexpand_dim
(t, dim, k)
fairseq/fairseq/modules/kmeans_attention.py:81
↓ 6 callersMethodforward
(self, src_tokens, src_lengths)
fairseq/tests/speech_recognition/asr_test_base.py:498
↓ 6 callersMethodfrom_pretrained
( cls, model_name_or_path, checkpoint_file="model.pt", data_name_or_path=".",
fairseq/fairseq/models/bart/model.py:117
↓ 6 callersMethodfrom_string
Parse the configuration of lambda coefficient (for scheduling). x = "3" # lambda will be a constant equal to x
fairseq/fairseq/tasks/online_backtranslation.py:60
↓ 6 callersMethodgcmvn_denormalize
(self, x)
fairseq/fairseq/speech_generator.py:21
↓ 6 callersMethodget_input
(self, batch, k)
models/taming/models/vqgan.py:75
↓ 6 callersFunctionget_padding
(kernel_size, dilation=1)
fairseq/fairseq/models/text_to_speech/hifigan.py:16
↓ 6 callersFunctionget_parser
(desc, default_task="translation")
fairseq/fairseq/options.py:221
↓ 6 callersMethodget_target_dictionary
(self, lang)
fairseq/fairseq/data/multilingual/multilingual_data_manager.py:441
↓ 6 callersFunctionget_trainer_and_epoch_itr
(epoch, epoch_size, num_updates, iterations_in_epoch)
fairseq/tests/test_train.py:38
↓ 6 callersFunctionget_waveform
Get the waveform and sample rate of a 16-bit WAV/FLAC/OGG Vorbis audio. Args: path_or_fp (str or BinaryIO): the path or file-like object
fairseq/fairseq/data/audio/audio_utils.py:64
↓ 6 callersMethodget_waveform
(self, feat)
fairseq/fairseq/speech_generator.py:31
↓ 6 callersFunctionget_world_size
(group)
fairseq/fairseq/distributed/utils.py:428
↓ 6 callersFunctionlang_pair_dataset
(lengths: Sequence[int])
fairseq/tests/test_dataset.py:14
↓ 6 callersFunctionload_langpair_dataset
( data_path, split, src, src_dict, tgt, tgt_dict, combine, dataset_impl, u
fairseq/fairseq/tasks/translation.py:39
↓ 6 callersFunctionmake_right_to_left
(line)
fairseq/examples/noisychannel/rerank_utils.py:241
↓ 6 callersMethodmax_positions
Return the max sentence length allowed by the task.
tasks/ofa_task.py:341
↓ 6 callersMethodmaybe_layer_norm
(self, layer_norm, x, before=False, after=False)
fairseq/fairseq/model_parallel/models/pipeline_parallel_transformer/layers.py:577
↓ 6 callersMethodmaybe_layer_norm
(self, layer_norm, x, before=False, after=False)
fairseq/fairseq/models/lightconv.py:870
↓ 6 callersFunctionmerge
(key, left_pad, move_eos_to_beginning=False, pad_to_length=None)
fairseq/fairseq/data/denoising_dataset.py:28
↓ 6 callersMethodnext_tokens
The set of child labels.
fairseq/fairseq/token_generation_constraints.py:141
↓ 6 callersFunctionprob_check
(tensor, eps=1e-10)
fairseq/examples/simultaneous_translation/utils/functions.py:9
↓ 6 callersMethodremove_weight_norm
(self)
fairseq/fairseq/models/text_to_speech/hifigan.py:104
↓ 6 callersMethodreset
(self, one_init=False)
fairseq/fairseq/scoring/bleu.py:101
↓ 6 callersMethodreverse
Load the model parameters from EMA model. Useful for inference or fine-tuning from the EMA model.
fairseq/fairseq/models/ema/ema.py:194
↓ 6 callersMethodrm
(path: str)
fairseq/fairseq/file_io.py:109
↓ 6 callersMethodsample
( self, sentences: List[str], beam: int = 1, verbose: bool = False, **kwargs )
fairseq/fairseq/hub_utils.py:126
↓ 6 callersMethodsave_file
(self, path_, split_type, data_type, lang)
fairseq/examples/multilingual/data_scripts/download_ted_and_extract.py:110
↓ 6 callersMethodset_lang_idx
(self, lang_idx)
fairseq/examples/latent_depth/latent_depth_src/models/latent_transformer.py:37
↓ 6 callersMethodsetup_task
Setup the task (e.g., load dictionaries).
fairseq/examples/speech_recognition/tasks/speech_recognition.py:101
↓ 6 callersMethodsize
(self)
utils/cider/pyciderevalcap/cider/cider_scorer.py:91
↓ 6 callersMethodsize
(self, index)
fairseq/fairseq/data/audio/speech_to_text_dataset.py:346
↓ 6 callersMethodsize
Return an example's size as a float or tuple. This value is used when filtering a dataset with ``--max-positions``.
fairseq/fairseq/data/audio/raw_audio_dataset.py:205
↓ 6 callersFunctionsplit_exists
(split, src, tgt, lang, data_path)
fairseq/fairseq/tasks/translation.py:62
↓ 6 callersMethodstep
One update of EMA which is done every self.config.ema_update_freq updates of the model. @param updates The current number of
fairseq/fairseq/models/ema/ema.py:169
↓ 6 callersMethodstep
Take a single search step. Args: step: the current search step, starting at 0 lprobs: (bsz x input_beam_size x vocab_
models/search.py:30
↓ 6 callersMethodstep_update
Called at the end of each step.
fairseq/fairseq/quantization_utils.py:118
↓ 6 callersFunctionto_gpu
(x)
fairseq/examples/textless_nlp/gslm/unit2speech/tacotron2/utils.py:53
↓ 6 callersFunctionto_np
(x)
fairseq/examples/speech_synthesis/generate_waveform.py:47
↓ 6 callersMethodtokenize
(cls, tokenizer, text: str)
fairseq/fairseq/data/audio/speech_to_text_dataset.py:226
↓ 6 callersFunctionupsample2
Upsampling the input by 2 using sinc interpolation. Smith, Julius, and Phil Gossett. "A flexible sampling-rate conversion method." ICASSP
fairseq/examples/speech_synthesis/preprocessing/denoiser/resample.py:35
↓ 6 callersFunctionwork_on_wmt
(directions, wmt_data)
fairseq/examples/multilingual/data_scripts/download_wmt19_and_before.py:839
↓ 5 callersFunctionEmbedding
(num_embeddings, embedding_dim, padding_idx)
fairseq/fairseq/models/lstm.py:666
↓ 5 callersMethod__init__
(self, args)
fairseq/fairseq/model_parallel/models/pipeline_parallel_transformer/layers.py:311
↓ 5 callersMethod__init__
(self, *args, **kwargs)
fairseq/fairseq/models/fairseq_model.py:357
↓ 5 callersMethod__init__
( self, embed_dim, n_heads, hidden_dim, kernel_size, dropout, attention_dropout )
fairseq/fairseq/models/text_to_speech/fastspeech2.py:58
↓ 5 callersMethod_get_input_buffer
( self, incremental_state: Optional[Dict[str, Dict[str, Optional[Tensor]]]] )
fairseq/examples/linformer/linformer_src/modules/multihead_linear_attention.py:432
↓ 5 callersMethod_map_index
(self, key, index)
fairseq/fairseq/data/round_robin_zip_datasets.py:46
← previousnext →201–300 of 6,938, ranked by callers