Path Lines of Code pytext/__init__.py 100 pytext/builtin_task.py 73 pytext/common/__init__.py 11 pytext/common/constants.py 289 pytext/common/utils.py 3 pytext/config/__init__.py 9 pytext/config/component.py 184 pytext/config/config_adapter.py 981 pytext/config/contextual_intent_slot.py 31 pytext/config/doc_classification.py 26 pytext/config/field_config.py 84 pytext/config/module_config.py 39 pytext/config/pair_classification.py 11 pytext/config/pytext_config.py 256 pytext/config/query_document_pairwise_ranking.py 10 pytext/config/serialize.py 236 pytext/config/utils.py 56 pytext/data/__init__.py 33 pytext/data/batch_sampler.py 166 pytext/data/bert_tensorizer.py 268 pytext/data/data.py 259 pytext/data/data_handler.py 555 pytext/data/data_structures/__init__.py 1 pytext/data/data_structures/annotation.py 508 pytext/data/data_structures/node.py 32 pytext/data/decoupled_data.py 230 pytext/data/dense_retrieval_tensorizer.py 127 pytext/data/disjoint_multitask_data.py 61 pytext/data/disjoint_multitask_data_handler.py 110 pytext/data/dynamic_pooling_batcher.py 127 pytext/data/featurizer/__init__.py 3 pytext/data/featurizer/featurizer.py 35 pytext/data/featurizer/simple_featurizer.py 56 pytext/data/masked_tensorizer.py 96 pytext/data/masked_util.py 175 pytext/data/packed_lm_data.py 102 pytext/data/pickleable_gpt2bpe_encoder/__init__.py 6 pytext/data/pickleable_gpt2bpe_encoder/pickleable_gpt2bpe_encoder.py 11 pytext/data/roberta_tensorizer.py 182 pytext/data/sources/__init__.py 15 pytext/data/sources/conllu.py 115 pytext/data/sources/data_source.py 218 pytext/data/sources/dense_retrieval.py 105 pytext/data/sources/pandas.py 62 pytext/data/sources/session.py 40 pytext/data/sources/squad.py 294 pytext/data/sources/tsv.py 300 pytext/data/squad_for_bert_tensorizer.py 384 pytext/data/squad_tensorizer.py 257 pytext/data/tensorizers.py 1787 pytext/data/token_tensorizer.py 281 pytext/data/tokenizers/__init__.py 18 pytext/data/tokenizers/tokenizer.py 280 pytext/data/utils.py 285 pytext/data/xlm_constants.py 84 pytext/data/xlm_dictionary.py 174 pytext/data/xlm_tensorizer.py 185 pytext/exporters/__init__.py 6 pytext/exporters/custom_exporters.py 100 pytext/exporters/exporter.py 157 pytext/fields/__init__.py 42 pytext/fields/char_field.py 73 pytext/fields/contextual_token_embedding_field.py 46 pytext/fields/dict_field.py 102 pytext/fields/field.py 274 pytext/fields/text_field_with_special_unk.py 107 pytext/legacy/__init__.py 6 pytext/legacy/data/__init__.py 49 pytext/legacy/data/batch.py 74 pytext/legacy/data/dataset.py 207 pytext/legacy/data/example.py 90 pytext/legacy/data/field.py 393 pytext/legacy/data/iterator.py 209 pytext/legacy/data/pipeline.py 33 pytext/legacy/datasets/__init__.py 55 pytext/legacy/datasets/babi.py 161 pytext/legacy/datasets/imdb.py 43 pytext/legacy/datasets/language_modeling.py 115 pytext/legacy/datasets/nli.py 181 pytext/legacy/datasets/sequence_tagging.py 76 pytext/legacy/datasets/sst.py 83 pytext/legacy/datasets/text_classification.py 177 pytext/legacy/datasets/translation.py 184 pytext/legacy/datasets/trec.py 51 pytext/legacy/datasets/unsupervised_learning.py 125 pytext/legacy/vocab.py 131 pytext/loss/__init__.py 60 pytext/loss/loss.py 426 pytext/loss/regularized_loss.py 243 pytext/loss/regularizer.py 82 pytext/loss/structured_loss.py 42 pytext/main.py 552 pytext/metric_reporters/__init__.py 41 pytext/metric_reporters/calibration_metric_reporter.py 44 pytext/metric_reporters/channel.py 240 pytext/metric_reporters/classification_metric_reporter.py 202 pytext/metric_reporters/compositional_metric_reporter.py 176 pytext/metric_reporters/compositional_utils.py 41 pytext/metric_reporters/dense_retrieval_metric_reporter.py 113 pytext/metric_reporters/disjoint_multitask_metric_reporter.py 77 pytext/metric_reporters/intent_slot_detection_metric_reporter.py 169 pytext/metric_reporters/language_model_metric_reporter.py 243 pytext/metric_reporters/mask_compositional.py 347 pytext/metric_reporters/mask_seq2seq_topk.py 167 pytext/metric_reporters/metric_reporter.py 202 pytext/metric_reporters/multi_span_qa_metric_reporter.py 273 pytext/metric_reporters/pairwise_ranking_metric_reporter.py 20 pytext/metric_reporters/regression_metric_reporter.py 15 pytext/metric_reporters/seq2seq_compositional.py 166 pytext/metric_reporters/seq2seq_metric_reporter.py 142 pytext/metric_reporters/seq2seq_utils.py 2 pytext/metric_reporters/squad_metric_reporter.py 390 pytext/metric_reporters/word_tagging_metric_reporter.py 251 pytext/metrics/__init__.py 835 pytext/metrics/calibration_metrics.py 80 pytext/metrics/dense_retrieval_metrics.py 11 pytext/metrics/intent_slot_metrics.py 340 pytext/metrics/language_model_metrics.py 12 pytext/metrics/mask_metrics.py 243 pytext/metrics/seq2seq_metrics.py 87 pytext/metrics/squad_metrics.py 14 pytext/models/__init__.py 9 pytext/models/bert_classification_models.py 272 pytext/models/bert_regression_model.py 47 pytext/models/crf.py 173 pytext/models/decoders/__init__.py 4 pytext/models/decoders/decoder_base.py 18 pytext/models/decoders/intent_slot_model_decoder.py 51 pytext/models/decoders/mlp_decoder.py 59 pytext/models/decoders/mlp_decoder_n_tower.py 95 pytext/models/decoders/mlp_decoder_query_response.py 37 pytext/models/decoders/mlp_decoder_tri_tower.py 132 pytext/models/decoders/mlp_decoder_two_tower.py 160 pytext/models/decoders/multilabel_decoder.py 43 pytext/models/disjoint_multitask_model.py 39 pytext/models/distributed_model.py 31 pytext/models/doc_model.py 462 pytext/models/embeddings/__init__.py 22 pytext/models/embeddings/char_embedding.py 61 pytext/models/embeddings/contextual_token_embedding.py 38 pytext/models/embeddings/dict_embedding.py 105 pytext/models/embeddings/embedding_base.py 14 pytext/models/embeddings/embedding_list.py 48 pytext/models/embeddings/int_single_category_embedding.py 77 pytext/models/embeddings/int_weighted_multi_category_embedding.py 126 pytext/models/embeddings/mlp_embedding.py 53 pytext/models/embeddings/scriptable_embedding_list.py 62 pytext/models/embeddings/word_embedding.py 115 pytext/models/embeddings/word_seq_embedding.py 123 pytext/models/ensembles/__init__.py 4 pytext/models/ensembles/bagging_doc_ensemble.py 13 pytext/models/ensembles/bagging_intent_slot_ensemble.py 62 pytext/models/ensembles/ensemble.py 54 pytext/models/joint_model.py 137 pytext/models/language_models/__init__.py 1 pytext/models/language_models/lmlstm.py 156 pytext/models/masked_lm.py 155 pytext/models/masking_utils.py 29 pytext/models/model.py 216 pytext/models/module.py 47 pytext/models/output_layers/__init__.py 25 pytext/models/output_layers/distance_output_layer.py 155 pytext/models/output_layers/doc_classification_output_layer.py 161 pytext/models/output_layers/doc_regression_output_layer.py 68 pytext/models/output_layers/intent_slot_output_layer.py 110 pytext/models/output_layers/lm_output_layer.py 60 pytext/models/output_layers/multi_label_classification_layer.py 100 pytext/models/output_layers/output_layer_base.py 42 pytext/models/output_layers/pairwise_ranking_output_layer.py 16 pytext/models/output_layers/squad_output_layer.py 245 pytext/models/output_layers/utils.py 29 pytext/models/output_layers/word_tagging_output_layer.py 197 pytext/models/pair_classification_model.py 239 pytext/models/qna/__init__.py 1 pytext/models/qna/bert_squad_qa.py 131 pytext/models/qna/dr_qa.py 207 pytext/models/query_document_pairwise_ranking_model.py 88 pytext/models/r3f_models.py 120 pytext/models/representations/__init__.py 1 pytext/models/representations/attention.py 57 pytext/models/representations/augmented_lstm.py 329 pytext/models/representations/bilstm.py 98 pytext/models/representations/bilstm_doc_attention.py 57 pytext/models/representations/bilstm_doc_slot_attention.py 87 pytext/models/representations/bilstm_slot_attn.py 47 pytext/models/representations/biseqcnn.py 75 pytext/models/representations/contextual_intent_slot_rep.py 47 pytext/models/representations/deepcnn.py 172 pytext/models/representations/docnn.py 42 pytext/models/representations/huggingface_bert_sentence_encoder.py 85 pytext/models/representations/huggingface_electra_sentence_encoder.py 77 pytext/models/representations/jointcnn_rep.py 42 pytext/models/representations/lightconv.py 182 pytext/models/representations/ordered_neuron_lstm.py 118 pytext/models/representations/pair_rep.py 88 pytext/models/representations/pass_through.py 10 pytext/models/representations/pooling.py 95 pytext/models/representations/pure_doc_attention.py 49 pytext/models/representations/representation_base.py 11 pytext/models/representations/seq_rep.py 51 pytext/models/representations/slot_attention.py 57 pytext/models/representations/sparse_transformer_sentence_encoder.py 71 pytext/models/representations/stacked_bidirectional_rnn.py 75 pytext/models/representations/traced_transformer_encoder.py 111 pytext/models/representations/transformer/__init__.py 41 pytext/models/representations/transformer/luna_attention.py 602 pytext/models/representations/transformer/luna_sentence_encoder.py 386 pytext/models/representations/transformer/multihead_linear_attention.py 153 pytext/models/representations/transformer/representation.py 44 pytext/models/representations/transformer/sentence_encoder.py 133 pytext/models/representations/transformer_sentence_encoder.py 141 pytext/models/representations/transformer_sentence_encoder_base.py 94 pytext/models/roberta.py 455 pytext/models/semantic_parsers/__init__.py 1 pytext/models/semantic_parsers/rnng/__init__.py 1 pytext/models/semantic_parsers/rnng/rnng_constant.py 4 pytext/models/semantic_parsers/rnng/rnng_data_structures.py 116 pytext/models/semantic_parsers/rnng/rnng_parser.py 436 pytext/models/seq_models/__init__.py 1 pytext/models/seq_models/attention.py 223 pytext/models/seq_models/base.py 56 pytext/models/seq_models/contextual_intent_slot.py 46 pytext/models/seq_models/conv_decoder.py 475 pytext/models/seq_models/conv_encoder.py 310 pytext/models/seq_models/conv_model.py 60 pytext/models/seq_models/light_conv.py 110 pytext/models/seq_models/mask_generator.py 413 pytext/models/seq_models/nar_length.py 147 pytext/models/seq_models/nar_modules.py 58 pytext/models/seq_models/nar_output_layer.py 42 pytext/models/seq_models/nar_seq2seq_model.py 238 pytext/models/seq_models/positional.py 153 pytext/models/seq_models/projection_layers.py 153 pytext/models/seq_models/rnn_decoder.py 321 pytext/models/seq_models/rnn_encoder.py 152 pytext/models/seq_models/rnn_encoder_decoder.py 47 pytext/models/seq_models/seq2seq_model.py 202 pytext/models/seq_models/seq2seq_output_layer.py 29 pytext/models/seq_models/seqnn.py 28 pytext/models/seq_models/utils.py 43 pytext/models/tri_tower_classification_model.py 234 pytext/models/two_tower_classification_model.py 232 pytext/models/utils.py 3 pytext/models/word_model.py 196 pytext/optimizer/__init__.py 19 pytext/optimizer/activations.py 15 pytext/optimizer/adabelief.py 169 pytext/optimizer/fairseq_fp16_utils.py 147 pytext/optimizer/fp16_optimizer.py 534 pytext/optimizer/lamb.py 83 pytext/optimizer/madgrad.py 139 pytext/optimizer/optimizers.py 81 pytext/optimizer/radam.py 83 pytext/optimizer/scheduler.py 324 pytext/optimizer/sparsifiers/__init__.py 3 pytext/optimizer/sparsifiers/blockwise_sparsifier.py 134 pytext/optimizer/sparsifiers/sparsifier.py 506 pytext/optimizer/swa.py 190 pytext/resources/__init__.py 2 pytext/resources/roberta.py 10 pytext/task/__init__.py 15 pytext/task/disjoint_multitask.py 203 pytext/task/new_task.py 377 pytext/task/nop_decorator.py 21 pytext/task/pytext_checkpoint_management.py 56 pytext/task/serialize.py 175 pytext/task/task.py 164 pytext/task/tasks.py 335 pytext/torchscript/__init__.py 1 pytext/torchscript/batchutils.py 354 pytext/torchscript/module.py 1706 pytext/torchscript/seq2seq/__init__.py 1 pytext/torchscript/seq2seq/beam_decode.py 177 pytext/torchscript/seq2seq/beam_search.py 104 pytext/torchscript/seq2seq/decoder.py 94 pytext/torchscript/seq2seq/encoder.py 41 pytext/torchscript/seq2seq/export_model.py 141 pytext/torchscript/seq2seq/scripted_seq2seq_generator.py 108 pytext/torchscript/seq2seq/seq2seq_rnn_decoder_utils.py 29 pytext/torchscript/tensorizer/__init__.py 22 pytext/torchscript/tensorizer/bert.py 109 pytext/torchscript/tensorizer/normalizer.py 55 pytext/torchscript/tensorizer/roberta.py 105 pytext/torchscript/tensorizer/tensorizer.py 180 pytext/torchscript/tensorizer/xlm.py 159 pytext/torchscript/tokenizer/__init__.py 14 pytext/torchscript/tokenizer/bpe.py 72 pytext/torchscript/tokenizer/tokenizer.py 37 pytext/torchscript/utils.py 271 pytext/torchscript/vocab.py 88 pytext/trainers/__init__.py 13 pytext/trainers/ensemble_trainer.py 24 pytext/trainers/hogwild_trainer.py 59 pytext/trainers/trainer.py 562 pytext/trainers/training_state.py 26 pytext/utils/__init__.py 24 pytext/utils/ascii_table.py 49 pytext/utils/config_utils.py 48 pytext/utils/cuda.py 31 pytext/utils/data.py 197 pytext/utils/distributed.py 64 pytext/utils/documentation.py 142 pytext/utils/file_io.py 23 pytext/utils/label.py 96 pytext/utils/lazy.py 75 pytext/utils/loss.py 90 pytext/utils/meter.py 23 pytext/utils/mobile_onnx.py 109 pytext/utils/model.py 35 pytext/utils/onnx.py 137 pytext/utils/path.py 27 pytext/utils/precision.py 35 pytext/utils/tensor.py 7 pytext/utils/test.py 11 pytext/utils/timing.py 187 pytext/utils/torch.py 23 pytext/utils/typing.py 5 pytext/utils/usage.py 16 pytext/workflow.py 436 readthedocs.yml 7 setup.py 17