MCPcopy Create free account

hub / github.com/XiangLi1999/PrefixTuning / functions

Functions3,725 in github.com/XiangLi1999/PrefixTuning

↓ 1 callersFunctionconvert_tf2_checkpoint_to_pytorch
(tf_checkpoint_path, config_path, pytorch_dump_path)
transformers/src/transformers/convert_bert_original_tf2_checkpoint_to_pytorch.py:193
↓ 1 callersFunctionconvert_tf_checkpoint_to_pytorch
(tf_hub_path, pytorch_dump_path, is_encoder_named_decoder, vocab_size, is_encoder)
transformers/src/transformers/convert_tf_hub_seq_to_seq_bert_to_pytorch.py:33
↓ 1 callersFunctionconvert_tf_checkpoint_to_pytorch
(tf_checkpoint_path, config_file, pytorch_dump_path, discriminator_or_generator)
transformers/src/transformers/convert_electra_original_tf_checkpoint_to_pytorch.py:29
↓ 1 callersFunctionconvert_tf_checkpoint_to_pytorch
(tf_checkpoint_path, bert_config_file, pytorch_dump_path)
transformers/src/transformers/convert_bert_original_tf_checkpoint_to_pytorch.py:29
↓ 1 callersFunctionconvert_tf_checkpoint_to_pytorch
(tf_checkpoint_path, albert_config_file, pytorch_dump_path)
transformers/src/transformers/convert_albert_original_tf_checkpoint_to_pytorch.py:29
↓ 1 callersFunctionconvert_tf_checkpoint_to_pytorch
(tf_checkpoint_path, config_file, pytorch_dump_path)
transformers/src/transformers/convert_lxmert_original_tf_checkpoint_to_pytorch.py:29
↓ 1 callersFunctionconvert_tf_checkpoint_to_pytorch
(tf_checkpoint_path, config_file, pytorch_dump_path)
transformers/src/transformers/convert_t5_original_tf_checkpoint_to_pytorch.py:29
↓ 1 callersFunctionconvert_tf_checkpoint_to_pytorch
(tf_checkpoint_path, mobilebert_config_file, pytorch_dump_path)
transformers/src/transformers/convert_mobilebert_original_tf_checkpoint_to_pytorch.py:12
↓ 1 callersMethodconvert_to_tensors
Convert the inner content to tensors. Args: tensor_type (:obj:`str` or :class:`~transformers.tokenization_utils_base.Ten
transformers/src/transformers/tokenization_utils_base.py:515
↓ 1 callersFunctionconvert_to_unicode
Converts `text` to Unicode (if it's not already), assuming UTF-8 input.
transformers/src/transformers/tokenization_flaubert.py:63
↓ 1 callersFunctionconvert_trax_checkpoint_to_pytorch
(trax_model_pkl_path, config_file, pytorch_dump_path)
transformers/src/transformers/convert_reformer_trax_checkpoint_to_pytorch.py:185
↓ 1 callersFunctioncount_trainable_parameters
(model)
seq2seq/callbacks.py:14
↓ 1 callersMethodcreate_examples_from_document
Creates examples for a single document.
transformers/src/transformers/data/datasets/language_modeling.py:1690
↓ 1 callersMethodcreate_examples_from_document
Creates examples for a single document.
transformers/src/transformers/data/datasets/language_modeling.py:1882
↓ 1 callersMethodcreate_features_from_example
Creates examples for a single document.
transformers/src/transformers/data/data_collator.py:1447
↓ 1 callersMethodcreate_mask
Creates causal attention mask. Float mask where 1.0 indicates masked, 0.0 indicates not-masked. Args: qlen: Sequence len
transformers/src/transformers/modeling_xlnet.py:964
↓ 1 callersMethodcreate_mask
Creates causal attention mask. Float mask where 1.0 indicates masked, 0.0 indicates not-masked. Args: qlen: TODO Lysandr
transformers/src/transformers/modeling_tf_xlnet.py:462
↓ 1 callersFunctioncreate_optimizer
Creates an optimizer with a learning rate schedule using a warmup phase followed by a linear decay. Args: init_lr (:obj:`float`):
transformers/src/transformers/optimization_tf.py:82
↓ 1 callersMethodcreate_optimizer_and_scheduler
Setup the optimizer and the learning rate scheduler. We provide a reasonable default that works well. If you want to use something e
transformers/src/transformers/trainer_tf.py:214
↓ 1 callersMethodcreate_optimizer_and_scheduler
Setup the optimizer and the learning rate scheduler. We provide a reasonable default that works well. If you want to use something e
transformers/src/transformers/trainer.py:426
↓ 1 callersMethodcreate_optimizer_and_scheduler
Setup the optimizer and the learning rate scheduler. We provide a reasonable default that works well. If you want to use something e
gpt2/trainer_prefix.py:481
↓ 1 callersFunctioncreate_position_ids_from_input_ids
Replace non-padding symbols with their position numbers. Position numbers begin at padding_idx+1. Padding symbols are ignored. This is modified fr
transformers/src/transformers/modeling_roberta.py:1323
↓ 1 callersFunctioncreate_position_ids_from_input_ids
Replace non-padding symbols with their position numbers. Position numbers begin at padding_idx+1. Padding symbols are ignored. This is modified fr
transformers/src/transformers/modeling_longformer.py:103
↓ 1 callersMethodcreate_position_ids_from_input_ids
Replace non-padding symbols with their position numbers. Position numbers begin at padding_idx+1. Padding symbols are ignored. This is modifie
transformers/src/transformers/modeling_tf_longformer.py:165
↓ 1 callersMethodcreate_position_ids_from_input_ids
Replace non-padding symbols with their position numbers. Position numbers begin at padding_idx+1. Padding symbols are ignored. This is modifie
transformers/src/transformers/modeling_tf_roberta.py:110
↓ 1 callersMethodcreate_position_ids_from_inputs_embeds
We are provided embeddings directly. We cannot infer which are padded so just generate sequential position ids. :param tf.Tensor input
transformers/src/transformers/modeling_tf_longformer.py:177
↓ 1 callersMethodcreate_position_ids_from_inputs_embeds
We are provided embeddings directly. We cannot infer which are padded so just generate sequential position ids. :param tf.Tensor input
transformers/src/transformers/modeling_tf_roberta.py:122
↓ 1 callersMethodcreate_position_ids_from_inputs_embeds
We are provided embeddings directly. We cannot infer which are padded so just generate sequential position ids. :param torch.Tensor i
transformers/src/transformers/modeling_roberta.py:126
↓ 1 callersMethodcreate_position_ids_from_inputs_embeds
We are provided embeddings directly. We cannot infer which are padded so just generate sequential position ids. :param torch.Tensor i
transformers/src/transformers/modeling_longformer.py:175
↓ 1 callersFunctioncreate_sinusoidal_embeddings
(n_pos, dim, out)
transformers/src/transformers/modeling_distilbert.py:75
↓ 1 callersFunctioncreate_sinusoidal_embeddings
(n_pos, dim, out)
transformers/src/transformers/modeling_xlm.py:79
↓ 1 callersFunctioncreate_tf_var
(tensor: np.ndarray, name: str, session: tf.Session)
transformers/src/transformers/convert_bert_pytorch_checkpoint_to_original_tf.py:69
↓ 1 callersMethodcreate_token_type_ids_from_sequences
Create the token type IDs corresponding to the sequences passed. `What are token type IDs? <../glossary.html#token-type-ids>`__
transformers/src/transformers/tokenization_utils_base.py:2337
↓ 1 callersMethodcross_att
( self, lang_input, lang_attention_mask, visn_input, visn_attention_ma
transformers/src/transformers/modeling_tf_lxmert.py:485
↓ 1 callersMethodcross_att
( self, lang_input, lang_attention_mask, visual_input, visual_attentio
transformers/src/transformers/modeling_lxmert.py:487
↓ 1 callersMethoddecode
Converts a sequence of ids in a string, using the tokenizer and vocabulary with options to remove special tokens and clean up tokeniz
transformers/src/transformers/tokenization_utils_base.py:2701
↓ 1 callersMethoddelete_obj
Call HF API to delete a file stored by user
transformers/src/transformers/hf_api.py:184
↓ 1 callersMethoddelete_token
Delete token. Do not fail if token does not exist.
transformers/src/transformers/hf_api.py:255
↓ 1 callersFunctiondetokenize_numbers
Inverts the operation of `tokenize_numbers`. This is replacing ' @,@ ' and ' @.@' by ',' and '.'. Args: text: A string where the
transformers/src/transformers/tokenization_transfo_xl.py:100
↓ 1 callersMethoddistributed_prediction_steps
(self, batch)
transformers/src/transformers/trainer_tf.py:461
↓ 1 callersMethoddistributed_training_steps
(self, batch)
transformers/src/transformers/trainer_tf.py:683
↓ 1 callersFunctiondownload_and_unzip
(url, dest_dir)
transformers/src/transformers/convert_marian_to_pytorch.py:684
↓ 1 callersMethodembed_answers
( self, input_ids, attention_mask=None, checkpoint_batch_size=-1, )
transformers/src/transformers/modeling_retribert.py:156
↓ 1 callersMethodembed_questions
( self, input_ids, attention_mask=None, checkpoint_batch_size=-1, )
transformers/src/transformers/modeling_retribert.py:142
↓ 1 callersMethodenforce_repetition_penalty_
Enforce the repetition penalty (from the `CTRL paper <https://arxiv.org/abs/1909.05858>`__).
transformers/src/transformers/generation_utils.py:59
↓ 1 callersFunctionensure_valid_input
Ensure input are presented in the correct order, without any None Args: model: The model used to forward the input data token
transformers/src/transformers/convert_graph_to_onnx.py:118
↓ 1 callersMethodestimate_tokens
Helper function to estimate the total number of tokens from the model inputs. Args: inputs (:obj:`dict`): The model inpu
transformers/src/transformers/modeling_utils.py:322
↓ 1 callersFunctioneval
(args, model=None)
seq2seq/finetune.py:802
↓ 1 callersMethodevaluate
Run evaluation and returns metrics. The calling script will be responsible for providing a method to compute metrics, as they are
transformers/src/transformers/trainer_tf.py:417
↓ 1 callersFunctionexpand_group_to_two_letter_codes
(grp_name)
transformers/src/transformers/convert_marian_to_pytorch.py:330
↓ 1 callersFunctionfill_with_neg_inf
FP16-compatible function that fills a input_ids with -inf.
transformers/src/transformers/modeling_bart.py:1187
↓ 1 callersFunctionfill_with_neg_inf
FP16-compatible function that fills a input_ids with -inf.
transformers/src/transformers/modeling_fsmt.py:874
↓ 1 callersFunctionfind_all_best_thresh
(main_eval, preds, exact_raw, f1_raw, na_probs, qid_to_has_ans)
transformers/src/transformers/data/metrics/squad_metrics.py:202
↓ 1 callersFunctionfind_model_file
(dest_dir)
transformers/src/transformers/convert_marian_to_pytorch.py:178
↓ 1 callersFunctionfind_vocab_file
(model_dir)
transformers/src/transformers/convert_marian_to_pytorch.py:483
↓ 1 callersMethodfloating_point_ops
For models that inherit from :class:`~transformers.PretrainedModel`, uses that method to compute the number of floating point operati
transformers/src/transformers/trainer.py:1552
↓ 1 callersMethodfloating_point_ops
For models that inherit from :class:`~transformers.PretrainedModel`, uses that method to compute the number of floating point operati
gpt2/trainer_prefix.py:1638
↓ 1 callersMethodformat_dict
(d)
transformers/src/transformers/commands/env.py:57
↓ 1 callersMethodforward
Input shape: Time(SeqLen) x Batch x Channel
transformers/src/transformers/modeling_fsmt.py:751
↓ 1 callersMethodfreeze_embeds
Freeze token embeddings and positional embeddings for bart, just token embeddings for t5.
seq2seq/finetune.py:472
↓ 1 callersMethodfrom_dict
Instantiates a :class:`~transformers.PretrainedConfig` from a Python dictionary of parameters. Args: config_dict (:obj:`
transformers/src/transformers/configuration_utils.py:386
↓ 1 callersMethodfrom_str
Creates an instance of the right subclass of :class:`~transformers.pipelines.PipelineDataFormat` depending on :obj:`format`.
transformers/src/transformers/pipelines.py:269
↓ 1 callersMethodfrom_type
(comp_type: str, *args, **kwargs)
transformers/src/transformers/convert_dpr_original_checkpoint_to_pytorch.py:30
↓ 1 callersMethodgenerate
Implements RAG token decoding. Args: input_ids (:obj:`torch.LongTensor` of shape :obj:`(batch_size, sequence_length)`, `
transformers/src/transformers/modeling_rag.py:1190
↓ 1 callersFunctiongeneric_train
( model, args: argparse.Namespace, early_stopping_callback=False, logger=True, # can pass Wan
seq2seq/lightning_base.py:721
↓ 1 callersMethodget_added_vocab
Returns the added tokens in the vocabulary as a dictionary of token to index. Returns: :obj:`Dict[str, int]`: The added
transformers/src/transformers/tokenization_utils.py:147
↓ 1 callersFunctionget_checkpoint_callback
Saves the best model by validation ROUGE2 score.
seq2seq/callbacks.py:87
↓ 1 callersMethodget_codes
(self, selected_sents)
transformers/src/transformers/trainer_prefix.py:786
↓ 1 callersMethodget_config
(self)
transformers/src/transformers/modeling_tf_utils.py:748
↓ 1 callersMethodget_config
(self)
transformers/src/transformers/optimization_tf.py:72
↓ 1 callersMethodget_dataset
(self, type_path)
seq2seq/finetune.py:280
↓ 1 callersMethodget_dataset
(self, type_path)
seq2seq/finetune.py:597
↓ 1 callersMethodget_doc_dicts
Returns a list of dictionaries, containing titles and text of the retrieved documents. Args: doc_ids (:obj:`np.ndarray`
transformers/src/transformers/retrieval_rag.py:48
↓ 1 callersFunctionget_early_stopping_callback
(metric, patience)
seq2seq/callbacks.py:110
↓ 1 callersMethodget_emb
(self, sent_lst, word_lst, num_layer, bos_tok, eos_tok)
transformers/src/transformers/data/datasets/language_modeling.py:1494
↓ 1 callersMethodget_embedding
Build sinusoidal embeddings. This matches the implementation in tensor2tensor, but differs slightly from the description in Section 3
transformers/src/transformers/modeling_fsmt.py:1184
↓ 1 callersMethodget_eval_dataloader
Returns the evaluation :class:`~torch.utils.data.DataLoader`. Will use no sampler if :obj:`self.eval_dataset` is a :obj:`torch.utils
transformers/src/transformers/trainer_prefix.py:490
↓ 1 callersMethodget_eval_dataloader
Returns the evaluation :class:`~torch.utils.data.DataLoader`. Will use no sampler if :obj:`self.eval_dataset` is a :obj:`torch.utils
transformers/src/transformers/trainer.py:369
↓ 1 callersMethodget_eval_dataloader
Returns the evaluation :class:`~torch.utils.data.DataLoader`. Will use no sampler if :obj:`self.eval_dataset` is a :obj:`torch.utils
gpt2/trainer_prefix.py:422
↓ 1 callersMethodget_eval_tfdataset
Returns the evaluation :class:`~tf.data.Dataset`. Args: eval_dataset (:class:`~tf.data.Dataset`, `optional`):
transformers/src/transformers/trainer_tf.py:151
↓ 1 callersMethodget_example_from_tensor_dict
See base class.
transformers/src/transformers/data/processors/glue.py:389
↓ 1 callersMethodget_extended_attention_mask
(self, attention_mask, input_shape, dtype)
transformers/src/transformers/modeling_tf_electra.py:481
↓ 1 callersMethodget_fixlen_iter
(self, start=0)
transformers/src/transformers/tokenization_transfo_xl.py:667
↓ 1 callersFunctionget_from_cache
Given a URL, look for the corresponding file in the local cache. If it's not there, download it. Then return the path to the cached file.
transformers/src/transformers/file_utils.py:798
↓ 1 callersMethodget_gold_init
(self, gpt2, sample_input)
gpt2/train_control.py:184
↓ 1 callersMethodget_head_mask
(self, head_mask)
transformers/src/transformers/modeling_tf_electra.py:502
↓ 1 callersMethodget_input_embeddings
(self)
transformers/src/transformers/modeling_t5.py:659
↓ 1 callersMethodget_labels
(self)
transformers/src/transformers/data/datasets/glue.py:149
↓ 1 callersMethodget_labels
See base class.
transformers/src/transformers/data/processors/glue.py:410
↓ 1 callersMethodget_lr_scheduler
(self)
seq2seq/lightning_base.py:201
↓ 1 callersMethodget_lr_scheduler
(self)
seq2seq/lightning_base.py:502
↓ 1 callersFunctionget_masks
Generate hidden states mask, and optionally an attention mask.
transformers/src/transformers/modeling_tf_flaubert.py:160
↓ 1 callersFunctionget_masks
Generate hidden states mask, and optionally an attention mask.
transformers/src/transformers/modeling_tf_xlm.py:85
↓ 1 callersMethodget_nll
(self, seq_logits, doc_scores, target, reduce_loss=False, epsilon=0.0, exclude_bos_score=False)
transformers/src/transformers/modeling_rag.py:912
↓ 1 callersMethodget_nll
(self, seq_logits, doc_scores, target, reduce_loss=False, epsilon=0.0)
transformers/src/transformers/modeling_rag.py:1438
↓ 1 callersFunctionget_old_bleu
(o)
transformers/src/transformers/convert_marian_to_pytorch.py:55
↓ 1 callersMethodget_output_embeddings
Returns the model's output embeddings. Returns: :obj:`nn.Module`: A torch module mapping hidden states to vocabulary.
transformers/src/transformers/modeling_utils.py:457
↓ 1 callersMethodget_output_embeddings
(self)
transformers/src/transformers/modeling_encoder_decoder.py:182
↓ 1 callersMethodget_output_embeddings
(self)
transformers/src/transformers/modeling_mobilebert.py:948
↓ 1 callersMethodget_output_embeddings
(self)
transformers/src/transformers/modeling_mobilebert.py:1068
← previousnext →801–900 of 3,725, ranked by callers