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
↓ 3 callersFunctionLinear
(in_features, out_features, bias=True)
fairseq/fairseq/models/transformer/transformer_decoder.py:448
↓ 3 callersFunctionLinear
Weight-normalized Linear layer (input: N x T x C)
fairseq/examples/laser/laser_src/laser_lstm.py:554
↓ 3 callersMethod__init__
(self, models)
fairseq/fairseq/sequence_generator.py:730
↓ 3 callersMethod__init__
(self, args)
fairseq/fairseq/modules/transformer_layer.py:170
↓ 3 callersMethod__init__
(self, index)
fairseq/fairseq/modules/downsampled_multihead_attention.py:292
↓ 3 callersMethod__init__
(self, args, encoder)
fairseq/fairseq/model_parallel/models/roberta/model.py:43
↓ 3 callersMethod__init__
(self, cfg: DictConfig, params, fp32_optimizer, fp32_params, **kwargs)
fairseq/fairseq/optim/fp16_optimizer.py:249
↓ 3 callersMethod__init__
(self, iterable, chunk_size)
fairseq/fairseq/data/iterators.py:497
↓ 3 callersMethod__init__
(self, dictionary, bpe_cont_marker="@@", bpe_end_marker=None)
fairseq/fairseq/data/noising.py:14
↓ 3 callersMethod__init__
(self, encoder, decoder)
fairseq/fairseq/models/fconv.py:69
↓ 3 callersMethod__init__
( self, dictionary, embed_dim=512, max_positions=1024, convolutions=((
fairseq/fairseq/models/fconv_self_att.py:176
↓ 3 callersMethod__init__
(self, encoder, decoder)
fairseq/fairseq/models/lstm.py:29
↓ 3 callersMethod__init__
(self, args, encoder)
fairseq/fairseq/models/roberta/model.py:44
↓ 3 callersMethod__init__
(self, encoder, decoder)
fairseq/fairseq/models/speech_to_text/berard.py:37
↓ 3 callersMethod__init__
(self, args, module)
fairseq/fairseq/models/speech_to_text/modules/augmented_memory_attention.py:382
↓ 3 callersMethod__init__
( self, n_fft: int, win_length: int, hop_length: int, n_iter: int, window_fn=torch.han
fairseq/fairseq/models/text_to_speech/vocoder.py:50
↓ 3 callersMethod__init__
(self, cfg: Wav2Vec2CtcConfig, w2v_encoder: BaseFairseqModel)
fairseq/fairseq/models/wav2vec/wav2vec2_asr.py:162
↓ 3 callersMethod__init__
(self, dictionary, model)
fairseq/examples/speech_recognition/w2l_decoder.py:247
↓ 3 callersMethod__init__
(self, encoder, eos_num, feat_dim, adapter_type="None", adapter_dim=0)
fairseq/examples/speech_text_joint_to_text/models/s2t_dualinputtransformer.py:36
↓ 3 callersMethod__init__
(self, wav2vec_enc, mbart_enc, adaptor, shared_layers)
fairseq/examples/speech_text_joint_to_text/models/s2t_dualinputxmtransformer.py:74
↓ 3 callersMethod__init__
(self, args)
fairseq/examples/simultaneous_translation/modules/monotonic_multihead_attention.py:458
↓ 3 callersMethod__init__
(self, args, dictionary, embed_tokens, num_logits=1)
fairseq/examples/latent_depth/latent_depth_src/models/latent_transformer.py:22
↓ 3 callersMethod__init__
(self, n_in_channels, n_mel_channels, n_layers, n_channels, kernel_size)
fairseq/examples/textless_nlp/gslm/unit2speech/glow.py:111
↓ 3 callersMethod__init__
(self, in_channels, out_channels, kernel_size=1, stride=1, padding=None, dilation=1, bias=Tru
fairseq/examples/textless_nlp/gslm/unit2speech/tacotron2/layers.py:23
↓ 3 callersMethod__init__
(self, dim_encoded, dim_output, keep_hidden, num_layers)
fairseq/examples/textless_nlp/gslm/speech2unit/pretrained/cpc_feature_reader.py:147
↓ 3 callersMethod__init__
(self, cfg: SegmentationConfig)
fairseq/examples/wav2vec/unsupervised/models/wav2vec_u.py:86
↓ 3 callersMethod__init__
(self, models)
models/sequence_generator.py:777
↓ 3 callersMethod__init__
( self, type, length, projection, embed_dim,
models/ofa/unify_transformer.py:105
↓ 3 callersMethod__init__
(self, drop_prob=None)
models/ofa/unify_transformer_layer.py:99
↓ 3 callersMethod__init__
(self, cfg: SpeechTextUnifyConfig, encoder, decoder)
models/ofa/ofa_speech.py:169
↓ 3 callersMethod__init__
(self, layers, zero_init_residual=False, groups=1, width_per_group=64, replace_stride_with_di
models/ofa/resnet.py:138
↓ 3 callersMethod__init__
(self, num_features, logdet=False, affine=True, allow_reverse_init=False)
models/taming/modules/util.py:11
↓ 3 callersMethod__init__
(self, use_dropout=True)
models/taming/modules/losses/lpips.py:13
↓ 3 callersMethod__len__
Denotes the total number of samples
run_scripts/image_gen/fid_score.py:58
↓ 3 callersMethod_auto_convert_to_abs_path
(self, x)
fairseq/fairseq/data/audio/data_cfg.py:31
↓ 3 callersMethod_backtranslation_dataset_helper
( self, remove_eos_from_input_src, remove_eos_from_output_src, )
fairseq/tests/test_backtranslation_dataset.py:34
↓ 3 callersMethod_build_optimizer
(self)
trainer.py:288
↓ 3 callersMethod_build_optimizer
(self)
fairseq/fairseq/trainer.py:286
↓ 3 callersMethod_clean_if_not_none
(self, var_list)
fairseq/fairseq/data/multilingual/sampled_multi_dataset.py:123
↓ 3 callersMethod_convert_src_tokens_to_tensor
( self, vocab: Dictionary, src_tokens: List[List[str]], append_eos: bool )
fairseq/tests/test_noising.py:122
↓ 3 callersMethod_copy_keys
copy the prefixed keys (decoder_embed_dim) to the DC fields: decoder.embed_dim
fairseq/fairseq/models/transformer/transformer_config.py:244
↓ 3 callersFunction_create_dummy_alignment_data
(filename_src, filename_tgt, filename)
fairseq/tests/utils.py:174
↓ 3 callersFunction_create_dummy_data
(filename)
fairseq/tests/test_binaries.py:1746
↓ 3 callersFunction_demucs
(pretrained, url, **kwargs)
fairseq/examples/speech_synthesis/preprocessing/denoiser/pretrained.py:22
↓ 3 callersMethod_generate
( self, sample: Dict[str, Dict[str, Tensor]], prefix_tokens: Optional[Tensor] = None,
fairseq/fairseq/sequence_generator.py:189
↓ 3 callersMethod_generate
( self, models, sample: Dict[str, Dict[str, Tensor]], prefix_tokens: Optional[
models/sequence_generator.py:209
↓ 3 callersMethod_get_dataset_and_sample_index
(self, idx: int)
fairseq/fairseq/data/concat_dataset.py:41
↓ 3 callersFunction_get_kaldi_fbank
Get mel-filter bank features via PyKaldi.
fairseq/fairseq/data/audio/audio_utils.py:111
↓ 3 callersMethod_get_relative_position
( self, input: Tensor, max_relative_position: int, left_context_length: int,
fairseq/fairseq/models/speech_to_text/modules/emformer.py:1049
↓ 3 callersFunction_get_torchaudio_fbank
Get mel-filter bank features via TorchAudio.
fairseq/fairseq/data/audio/audio_utils.py:135
↓ 3 callersFunction_lang_token
(lang: str)
fairseq/fairseq/tasks/online_backtranslation.py:663
↓ 3 callersMethod_log_oom
(self, exc)
trainer.py:1301
↓ 3 callersMethod_log_oom
(self, exc)
fairseq/fairseq/trainer.py:1279
↓ 3 callersMethod_make_layer
(self, block, planes, blocks, stride=1, dilate=False, drop_path_rate=0.0)
models/ofa/resnet.py:185
↓ 3 callersFunction_override_attr
( sub_node: str, data_class: Type[FairseqDataclass], args: Namespace )
fairseq/fairseq/dataclass/utils.py:219
↓ 3 callersMethod_prepare_sample
(self, sample, is_dummy=False)
trainer.py:1241
↓ 3 callersMethod_prepare_sample
(self, sample, is_dummy=False)
fairseq/fairseq/trainer.py:1219
↓ 3 callersMethod_prepend_lang_bos_to_target
( self, dataset: LanguagePairDataset, lang: str )
fairseq/fairseq/tasks/online_backtranslation.py:259
↓ 3 callersMethod_reduce_and_log_stats
(self, logging_outputs, sample_size, grad_norm=None)
trainer.py:1430
↓ 3 callersMethod_reduce_and_log_stats
(self, logging_outputs, sample_size, grad_norm=None)
fairseq/fairseq/trainer.py:1408
↓ 3 callersMethod_set_decay
(self, decay)
fairseq/fairseq/models/ema/ema.py:128
↓ 3 callersFunction_symbols_to_sequence
(symbols)
fairseq/examples/textless_nlp/gslm/unit2speech/tacotron2/text.py:98
↓ 3 callersMethod_sync_fp16_grads_to_fp32
(self)
fairseq/fairseq/optim/fp16_optimizer.py:108
↓ 3 callersMethod_test_pretrained_masked_lm_for_translation
(self, learned_pos_emb, encoder_only)
fairseq/tests/test_binaries.py:1468
↓ 3 callersMethod_test_resume_training
(self, extra_clargs, arch="fconv_iwslt_de_en")
fairseq/tests/gpu/test_binaries_gpu.py:68
↓ 3 callersFunction_torch_persistent_save
(obj, f)
fairseq/fairseq/checkpoint_utils.py:513
↓ 3 callersMethod_train_transformer
(self, seed, extra_args=None)
fairseq/tests/test_checkpoint_utils.py:32
↓ 3 callersFunctionadd_common_eval_args
(group)
fairseq/fairseq/options.py:338
↓ 3 callersMethodadd_factory
(self, name, factory: Callable[[], Any])
fairseq/fairseq/tasks/fairseq_task.py:30
↓ 3 callersMethodadd_insertion_noise
(self, tokens, p)
fairseq/fairseq/data/denoising_dataset.py:372
↓ 3 callersFunctionall_reduce
(tensor, group, op="sum")
fairseq/fairseq/distributed/utils.py:511
↓ 3 callersMethodapply_mask
(self, x, padding_mask, target_list)
fairseq/fairseq/models/hubert/hubert.py:323
↓ 3 callersMethodassertAlmostEqual
(self, t1, t2)
fairseq/tests/test_label_smoothing.py:117
↓ 3 callersMethodassertAlmostEqual
(self, t1, t2)
fairseq/tests/distributed/test_bmuf.py:201
↓ 3 callersMethodassertTorchAllClose
(self, x, y, atol=1e-8, rtol=1e-5, msg=None)
fairseq/tests/test_ema.py:41
↓ 3 callersMethodassertTorchAllClose
(self, x, y, atol=1e-8, rtol=1e-5, msg=None)
fairseq/tests/gpu/test_ema_gpu.py:42
↓ 3 callersMethodassert_eos_at_end
Asserts last token of every sentence in x is EOS
fairseq/tests/test_noising.py:140
↓ 3 callersMethodasync_close
Wait for files to be written and clean up asynchronous PathManager. NOTE: `PathManager.async_close()` must be called at the end of an
fairseq/fairseq/file_io.py:185
↓ 3 callersFunctionaverage_checkpoints
Loads checkpoints from inputs and returns a model with averaged weights. Args: inputs: An iterable of string paths of checkpoints to load f
fairseq/scripts/average_checkpoints.py:16
↓ 3 callersFunctionbase_architecture
(args)
fairseq/fairseq/models/masked_lm.py:315
↓ 3 callersFunctionbase_architecture
(args)
fairseq/fairseq/models/lstm.py:699
↓ 3 callersFunctionbase_architecture
(args)
fairseq/examples/linformer/linformer_src/models/linformer_roberta.py:104
↓ 3 callersFunctionbase_lm_architecture
(args)
fairseq/fairseq/model_parallel/models/transformer_lm.py:91
↓ 3 callersFunctionbase_lm_architecture
(args)
fairseq/fairseq/models/fconv_lm.py:96
↓ 3 callersFunctionberard
The original version: "End-to-End Automatic Speech Translation of Audiobooks" (https://arxiv.org/abs/1802.04200)
fairseq/fairseq/models/speech_to_text/berard.py:547
↓ 3 callersFunctionbinarize
( databin_dir, direction, spm_vocab=SPM_VOCAB, prefix='', splits=['train', 'test', 'valid'], p
fairseq/examples/multilingual/data_scripts/binarize.py:117
↓ 3 callersMethodbinarize
(self, s: str, append_eos: bool = False)
fairseq/examples/roberta/wsc/wsc_task.py:85
↓ 3 callersFunctionbinarize_
( bpe_dir, databin_dir, direction, spm_vocab=SPM_VOCAB, splits=['train', 'test', 'valid'], )
fairseq/examples/multilingual/data_scripts/binarize.py:68
↓ 3 callersMethodbuffered_future_mask
(self, tensor)
fairseq/examples/speech_recognition/models/vggtransformer.py:700
↓ 3 callersMethodbuffered_sparse_mask
(self, tensor, tgt_len, src_len)
fairseq/fairseq/modules/sparse_multihead_attention.py:119
↓ 3 callersMethodbuild_dataset_for_inference
(self, src_tokens, src_lengths, sort=True)
fairseq/fairseq/tasks/masked_lm.py:222
↓ 3 callersFunctionbuild_dictionary
(filenames, src=False, tgt=False)
fairseq/fairseq_cli/preprocess.py:63
↓ 3 callersMethodbuild_embedding
(cls, cfg, dictionary, embed_dim, path=None)
fairseq/fairseq/models/transformer/transformer_base.py:105
↓ 3 callersMethodbuild_embedding
(cls, args, dictionary, embed_dim, path=None)
models/ofa/unify_transformer.py:453
↓ 3 callersMethodbuild_shared_embeddings
Helper function to build shared embeddings for a set of languages after checking that all dicts corresponding to those languages are
fairseq/fairseq/models/fairseq_model.py:385
↓ 3 callersFunctionbyte_decode
(x: str)
fairseq/fairseq/data/encoders/byte_utils.py:27
↓ 3 callersFunctionbyte_encode
(x: str)
fairseq/fairseq/data/encoders/byte_utils.py:22
↓ 3 callersMethodclassification_forward
(self, x)
fairseq/examples/discriminative_reranking_nmt/models/discriminative_reranking_model.py:273
↓ 3 callersMethodclip_grad_norm
(self, clip_norm)
fairseq/fairseq/trainer.py:1170
↓ 3 callersFunctioncollapse_whitespace
(text)
fairseq/examples/textless_nlp/gslm/unit2speech/tacotron2/cleaners.py:60
← previousnext →501–600 of 6,938, ranked by callers