pytext.torchscript package

Subpackages

• pytext.torchscript.tensorizer package
  • Submodules
  • pytext.torchscript.tensorizer.bert module
  • pytext.torchscript.tensorizer.normalizer module
  • pytext.torchscript.tensorizer.roberta module
  • pytext.torchscript.tensorizer.tensorizer module
  • pytext.torchscript.tensorizer.xlm module
  • Module contents
• pytext.torchscript.tokenizer package
  • Submodules
  • pytext.torchscript.tokenizer.bpe module
  • pytext.torchscript.tokenizer.tokenizer module
  • Module contents

Submodules

pytext.torchscript.module module

class pytext.torchscript.module.ScriptModule(optimize=None, _qualified_name=None, _compilation_unit=None, _cpp_module=None)

Bases: torch.jit.ScriptModule

class pytext.torchscript.module.ScriptTextModule(model: torch.jit.ScriptModule, output_layer: torch.jit.ScriptModule, tensorizer: pytext.torchscript.tensorizer.tensorizer.ScriptTensorizer)

Bases: pytext.torchscript.module.ScriptModule

class pytext.torchscript.module.ScriptTokenLanguageModule(model: torch.jit.ScriptModule, output_layer: torch.jit.ScriptModule, tensorizer: pytext.torchscript.tensorizer.tensorizer.ScriptTensorizer)

Bases: pytext.torchscript.module.ScriptModule

class pytext.torchscript.module.ScriptTokenLanguageModuleWithDenseFeature(model: torch.jit.ScriptModule, output_layer: torch.jit.ScriptModule, tensorizer: pytext.torchscript.tensorizer.tensorizer.ScriptTensorizer)

Bases: pytext.torchscript.module.ScriptModule

class pytext.torchscript.module.ScriptTokenModule(model: torch.jit.ScriptModule, output_layer: torch.jit.ScriptModule, tensorizer: pytext.torchscript.tensorizer.tensorizer.ScriptTensorizer)

Bases: pytext.torchscript.module.ScriptModule

pytext.torchscript.module.get_script_module_cls(input_type: pytext.torchscript.utils.ScriptInputType) → torch.jit.ScriptModule

pytext.torchscript.utils module

class pytext.torchscript.utils.ScriptInputType

Bases: enum.Enum

An enumeration.

is_text()

is_token()

text = 1

token = 2

pytext.torchscript.vocab module

class pytext.torchscript.vocab.ScriptVocabulary(vocab_list, unk_idx: int = 0, pad_idx: int = -1, bos_idx: int = -1, eos_idx: int = -1)

Bases: torch.jit.ScriptModule

Module contents