MCPcopy Create free account

hub / github.com/XiangLi1999/PrefixTuning / functions

Functions3,725 in github.com/XiangLi1999/PrefixTuning

↓ 2 callersMethod_use_cache
During generation, decide whether to pass the `past` variable to the next forward pass.
transformers/src/transformers/generation_tf_utils.py:39
↓ 2 callersFunctionacc_and_f1
(preds, labels)
transformers/src/transformers/data/metrics/__init__.py:36
↓ 2 callersMethodadd
Add a new hypothesis to the list.
transformers/src/transformers/generation_tf_utils.py:1070
↓ 2 callersMethodadd_examples
( self, texts_or_text_and_labels, labels=None, ids=None, overwrite_labels=False, overwrite_examples=Fa
transformers/src/transformers/data/processors/utils.py:194
↓ 2 callersFunctionadd_generic_args
(parser, root_dir)
seq2seq/lightning_base.py:678
↓ 2 callersMethodadd_model_specific_args
(parser, root_dir)
seq2seq/finetune.py:656
↓ 2 callersMethodadd_symbol
(self, sym)
transformers/src/transformers/tokenization_transfo_xl.py:349
↓ 2 callersFunctionapply_no_ans_threshold
(scores, na_probs, qid_to_has_ans, na_prob_thresh)
transformers/src/transformers/data/metrics/squad_metrics.py:97
↓ 2 callersFunctionapply_print_resets
(buf)
transformers/src/transformers/testing_utils.py:210
↓ 2 callersFunctionassert_all_frozen
(model)
seq2seq/utils.py:414
↓ 2 callersMethodbpe
(self, token)
transformers/src/transformers/tokenization_openai.py:133
↓ 2 callersMethodbpe
(self, token)
transformers/src/transformers/tokenization_xlm.py:700
↓ 2 callersMethodbuild
(self, input_shape)
transformers/src/transformers/modeling_tf_mobilebert.py:98
↓ 2 callersMethodbuild
Build shared word embedding layer
transformers/src/transformers/modeling_tf_funnel.py:88
↓ 2 callersFunctionbuild_shape_dict
(name: str, tensor, is_input: bool, seq_len: int)
transformers/src/transformers/convert_graph_to_onnx.py:159
↓ 2 callersFunctioncalc_banned_bad_words_ids
(prev_input_ids, bad_words_ids)
transformers/src/transformers/generation_tf_utils.py:931
↓ 2 callersFunctioncalc_banned_ngram_tokens
(prev_input_ids, num_hypos, no_repeat_ngram_size, cur_len)
transformers/src/transformers/generation_tf_utils.py:908
↓ 2 callersFunctioncalculate_rouge
(output_lns: List[str], reference_lns: List[str], use_stemmer=True)
seq2seq/utils.py:385
↓ 2 callersMethodcall
( self, hidden_states, attention_mask=None, position_bias=None, encode
transformers/src/transformers/modeling_tf_t5.py:424
↓ 2 callersFunctioncast_marian_config
(raw_cfg: Dict[str, str])
transformers/src/transformers/convert_marian_to_pytorch.py:163
↓ 2 callersMethodclean_up_tokenization
Clean up a list of simple English tokenization artifacts like spaces before punctuations and abreviated forms. Args: out
transformers/src/transformers/tokenization_utils_base.py:2755
↓ 2 callersMethodcompute_loss
How the loss is computed by Trainer. By default, all models return the loss in the first element. Subclass and override for custom b
transformers/src/transformers/trainer_prefix.py:2529
↓ 2 callersMethodcompute_loss
How the loss is computed by Trainer. By default, all models return the loss in the first element. Subclass and override for custom b
gpt2/trainer_prefix.py:1201
↓ 2 callersMethodcompute_loss_ADAP
How the loss is computed by Trainer. By default, all models return the loss in the first element. Subclass and override for custom b
transformers/src/transformers/trainer_prefix.py:2426
↓ 2 callersMethodcompute_loss_amortized_pplm
How the loss is computed by Trainer. By default, all models return the loss in the first element. Subclass and override for custom b
transformers/src/transformers/trainer_prefix.py:3438
↓ 2 callersMethodcompute_loss_distill
How the loss is computed by Trainer. By default, all models return the loss in the first element. Subclass and override for custom b
transformers/src/transformers/trainer_prefix.py:2554
↓ 2 callersMethodcompute_loss_distill
How the loss is computed by Trainer. By default, all models return the loss in the first element. Subclass and override for custom b
gpt2/trainer_prefix.py:1226
↓ 2 callersFunctionconvert_gpt2_checkpoint_to_pytorch
(gpt2_checkpoint_path, gpt2_config_file, pytorch_dump_folder_path)
transformers/src/transformers/convert_gpt2_original_tf_checkpoint_to_pytorch.py:29
↓ 2 callersFunctionconvert_openai_checkpoint_to_pytorch
(openai_checkpoint_folder_path, openai_config_file, pytorch_dump_folder_path)
transformers/src/transformers/convert_openai_original_tf_checkpoint_to_pytorch.py:29
↓ 2 callersFunctionconvert_opus_name_to_hf_name
(x)
transformers/src/transformers/convert_marian_to_pytorch.py:223
↓ 2 callersFunctionconvert_tf_weight_name_to_pt_weight_name
Convert a TF 2.0 model variable name in a pytorch model weight name. Conventions for TF2.0 scopes -> PyTorch attribute names conversions:
transformers/src/transformers/modeling_tf_pytorch_utils.py:30
↓ 2 callersMethodconvert_tokens_to_string
Converts a sequence of token ids in a single string. The most simple way to do it is ``" ".join(tokens)`` but we often want to remov
transformers/src/transformers/tokenization_utils.py:728
↓ 2 callersFunctionconvert_transfo_xl_checkpoint_to_pytorch
( tf_checkpoint_path, transfo_xl_config_file, pytorch_dump_folder_path, transfo_xl_dataset_file )
transformers/src/transformers/convert_transfo_xl_original_tf_checkpoint_to_pytorch.py:47
↓ 2 callersFunctionconvert_xlm_checkpoint_to_pytorch
(xlm_checkpoint_path, pytorch_dump_folder_path)
transformers/src/transformers/convert_xlm_original_pytorch_checkpoint_to_pytorch.py:32
↓ 2 callersFunctionconvert_xlnet_checkpoint_to_pytorch
( tf_checkpoint_path, bert_config_file, pytorch_dump_folder_path, finetuning_task=None )
transformers/src/transformers/convert_xlnet_original_tf_checkpoint_to_pytorch.py:51
↓ 2 callersMethodcreate_from_csv
( cls, file_name, split_name="", column_label=0, column_text=1, column_id=None, skip_first_row=False,
transformers/src/transformers/data/processors/utils.py:142
↓ 2 callersMethodcreate_sample
QuestionAnsweringPipeline leverages the :class:`~transformers.SquadExample` internally. This helper method encapsulate all the logic
transformers/src/transformers/pipelines.py:1599
↓ 2 callersMethodcreate_token_type_ids_from_sequences
Create a mask from the two sequences passed to be used in a sequence-pair classification task. An XLM sequence pair mask has the foll
transformers/src/transformers/tokenization_xlm.py:911
↓ 2 callersMethoddetokenize
Detokenize the provided tokens ids to readable text: - **tokens_ids**: List of tokens ids - **skip_special_tokens**: Flag ind
transformers/src/transformers/commands/serving.py:181
↓ 2 callersMethoddevice
(self)
transformers/src/transformers/benchmark/benchmark_args.py:104
↓ 2 callersMethodembed_sentences_checkpointed
( self, input_ids, attention_mask, sent_encoder, checkpoint_batch_size
transformers/src/transformers/modeling_retribert.py:98
↓ 2 callersFunctionencode_line
Only used by LegacyDataset
seq2seq/utils.py:54
↓ 2 callersMethodensure_exactly_one_mask_token
(self, masked_index: np.ndarray)
transformers/src/transformers/pipelines.py:1177
↓ 2 callersMethodevaluate
Run evaluation and returns metrics. The calling script will be responsible for providing a method to compute metrics, as they are
gpt2/trainer_prefix.py:1400
↓ 2 callersFunctionfind_best_thresh
(preds, scores, na_probs, qid_to_has_ans)
transformers/src/transformers/data/metrics/squad_metrics.py:179
↓ 2 callersFunctionfind_best_thresh_v2
(preds, scores, na_probs, qid_to_has_ans)
transformers/src/transformers/data/metrics/squad_metrics.py:134
↓ 2 callersFunctionflatten_list
(summary_ids: List[List])
seq2seq/utils.py:351
↓ 2 callersMethodforward
Input shape: Time(SeqLen) x Batch x Channel
transformers/src/transformers/modeling_bart.py:739
↓ 2 callersMethodfrom_encoder_decoder_configs
r""" Instantiate a :class:`~transformers.EncoderDecoderConfig` (or a derived class) from a pre-trained encoder model configuration and decoder
transformers/src/transformers/configuration_encoder_decoder.py:91
↓ 2 callersMethodfrom_question_encoder_generator_configs
r""" Instantiate a :class:`~transformers.EncoderDecoderConfig` (or a derived class) from a pre-trained encoder model configuration and decoder
transformers/src/transformers/configuration_rag.py:158
↓ 2 callersFunctiongenerate_identified_filename
Append a string-identifier at the end (before the extension, if any) to the provided filepath. Args: filename: pathlib.Path The actu
transformers/src/transformers/convert_graph_to_onnx.py:78
↓ 2 callersMethodgenerator
(self)
transformers/src/transformers/modeling_rag.py:1029
↓ 2 callersMethodget_batch
(self, i, bptt=None)
transformers/src/transformers/tokenization_transfo_xl.py:651
↓ 2 callersMethodget_char_lens
(data_file)
seq2seq/utils.py:130
↓ 2 callersFunctionget_cpu_memory
measures current cpu memory usage of a given `process_id` Args: - `process_id`: (`int`) process_id
transformers/src/transformers/benchmark/benchmark_utils.py:206
↓ 2 callersMethodget_dev_examples
See base class.
transformers/src/transformers/data/processors/glue.py:402
↓ 2 callersFunctionget_final_text
Project the tokenized prediction back to the original text.
transformers/src/transformers/data/metrics/squad_metrics.py:243
↓ 2 callersFunctionget_framework
Select framework (TensorFlow or PyTorch) to use. Args: model (:obj:`str`, :class:`~transformers.PreTrainedModel` or :class:`~transfo
transformers/src/transformers/pipelines.py:88
↓ 2 callersMethodget_input_embeddings
Returns the model's input embeddings. Returns: :obj:`tf.keras.layers.Layer`: A torch module mapping vocabulary to hidden
transformers/src/transformers/modeling_tf_utils.py:262
↓ 2 callersMethodget_labels
Gets the list of labels for this data set.
transformers/src/transformers/data/processors/utils.py:106
↓ 2 callersMethodget_length_logits
(self, input_sent, len_criterion)
transformers/src/transformers/trainer_prefix.py:2203
↓ 2 callersFunctionget_masks
Generate hidden states mask, and optionally an attention mask.
transformers/src/transformers/modeling_xlm.py:87
↓ 2 callersMethodget_output_embeddings
(self)
transformers/src/transformers/modeling_t5.py:662
↓ 2 callersFunctionget_pairs
Return set of symbol pairs in a word. word is represented as tuple of symbols (symbols being variable-length strings)
transformers/src/transformers/tokenization_openai.py:47
↓ 2 callersFunctionget_pairs
Return set of symbol pairs in a word. Word is represented as tuple of symbols (symbols being variable-length strings).
transformers/src/transformers/tokenization_phobert.py:52
↓ 2 callersFunctionget_pairs
Return set of symbol pairs in a word. Word is represented as tuple of symbols (symbols being variable-length strings).
transformers/src/transformers/tokenization_bertweet.py:52
↓ 2 callersFunctionget_pairs
Return set of symbol pairs in a word. Word is represented as tuple of symbols (symbols being variable-length strings).
transformers/src/transformers/tokenization_gpt2.py:91
↓ 2 callersFunctionget_pairs
Return set of symbol pairs in a word. Word is represented as tuple of symbols (symbols being variable-length strings).
transformers/src/transformers/tokenization_ctrl.py:102
↓ 2 callersFunctionget_pairs
Return set of symbol pairs in a word. word is represented as tuple of symbols (symbols being variable-length strings)
transformers/src/transformers/tokenization_xlm.py:430
↓ 2 callersFunctionget_pairs
Return set of symbol pairs in a word. word is represented as tuple of symbols (symbols being variable-length strings)
transformers/src/transformers/tokenization_fsmt.py:45
↓ 2 callersMethodget_prompt_p5
(self, control_code=None, gpt2=None, bsz=None, sample_size=1)
seq2seq/prefixTuning.py:412
↓ 2 callersFunctionget_tags
(code, ref_name)
transformers/src/transformers/convert_marian_to_pytorch.py:339
↓ 2 callersFunctionget_tokens
(s)
transformers/src/transformers/data/metrics/squad_metrics.py:45
↓ 2 callersMethodget_train_examples
See base class.
transformers/src/transformers/data/processors/glue.py:398
↓ 2 callersMethodget_vocab
Returns the vocabulary as a dictionary of token to index. :obj:`tokenizer.get_vocab()[token]` is equivalent to :obj:`tokenizer.conve
transformers/src/transformers/tokenization_utils_fast.py:88
↓ 2 callersFunctionglue_convert_examples_to_features
Loads a data file into a list of ``InputFeatures`` Args: examples: List of ``InputExamples`` or ``tf.data.Dataset`` containing the e
transformers/src/transformers/data/processors/glue.py:35
↓ 2 callersMethodhandle_args
(args: Sequence[Any])
transformers/src/transformers/pipelines.py:155
↓ 2 callersMethodids_to_clean_text
(self, generated_ids: List[int])
seq2seq/finetune.py:146
↓ 2 callersMethodids_to_clean_text
(self, generated_ids: List[int])
seq2seq/finetune.py:491
↓ 2 callersFunctioninfer_shapes
Attempt to infer the static vs dynamic axes for each input and output tensors for a specific model. Args: nlp: The pipeline object ho
transformers/src/transformers/convert_graph_to_onnx.py:145
↓ 2 callersMethodinit_attention_inputs
Returns the attention inputs associated to the inputs of the model.
transformers/src/transformers/modeling_funnel.py:190
↓ 2 callersMethodinit_attention_inputs
Returns the attention inputs associated to the inputs of the model.
transformers/src/transformers/modeling_tf_funnel.py:176
↓ 2 callersMethodinit_mems
(self, bsz)
transformers/src/transformers/modeling_tf_transfo_xl.py:469
↓ 2 callersMethodinit_mems
(self, bsz)
transformers/src/transformers/modeling_transfo_xl.py:797
↓ 2 callersFunctionis_sklearn_available
()
transformers/src/transformers/data/metrics/__init__.py:27
↓ 2 callersMethodis_world_master
Whether or not this process is the global main process (when training in a distributed fashion on several machines, this is only goin
transformers/src/transformers/trainer_prefix.py:2610
↓ 2 callersMethodis_world_master
Whether or not this process is the global main process (when training in a distributed fashion on several machines, this is only goin
gpt2/trainer_prefix.py:1282
↓ 2 callersFunctionlabel_smoothed_nll_loss
From fairseq
seq2seq/utils.py:33
↓ 2 callersFunctionload_layers_
(layer_lst: torch.nn.ModuleList, opus_state: dict, converter, is_decoder=False)
transformers/src/transformers/convert_marian_to_pytorch.py:122
↓ 2 callersFunctionload_pytorch_checkpoint_in_tf2_model
Load pytorch checkpoints in a TF 2.0 model
transformers/src/transformers/modeling_tf_pytorch_utils.py:74
↓ 2 callersFunctionload_pytorch_weights_in_tf2_model
Load pytorch state_dict in a TF 2.0 model.
transformers/src/transformers/modeling_tf_pytorch_utils.py:106
↓ 2 callersFunctionload_vocab
Loads a vocabulary file into a dictionary.
transformers/src/transformers/tokenization_bert.py:100
↓ 2 callersFunctionload_yaml
(path)
transformers/src/transformers/convert_marian_to_pytorch.py:716
↓ 2 callersMethodlog
Log :obj:`logs` on the various objects watching training. Subclass and override this method to inject custom behavior. Args
gpt2/trainer_prefix.py:1053
↓ 2 callersFunctionmain
()
gpt2/run_language_modeling.py:493
↓ 2 callersMethodmake_dynamic_sampler
(self, max_tokens_per_batch=1024, **kwargs)
seq2seq/utils.py:144
↓ 2 callersFunctionmake_registry
(repo_path="Opus-MT-train/models")
transformers/src/transformers/convert_marian_to_pytorch.py:368
↓ 2 callersMethodmake_weight
(self, num_positions, embedding_dim, padding_idx)
transformers/src/transformers/modeling_fsmt.py:1171
↓ 2 callersMethodmarginalize
(self, seq_logits, doc_scores)
transformers/src/transformers/modeling_rag.py:1060
↓ 2 callersFunctionmeasure_peak_memory_cpu
measures peak cpu memory consumption of a given `function` running the function for at least interval seconds and at most 20 * interva
transformers/src/transformers/benchmark/benchmark_utils.py:183
← previousnext →401–500 of 3,725, ranked by callers