PyTextConfig

class pytext.config.pytext_config.PyTextConfig

Bases: ConfigBase

All Attributes (including base classes)

task: Union[TaskBase.Config, Task_Deprecated.Config, _NewTask.Config, NewTask.Config, DisjointMultitask.Config, NewDisjointMultitask.Config, QueryDocumentPairwiseRankingTask.Config, EnsembleTask.Config, DocumentClassificationTask.Config, DocumentRegressionTask.Config, NewBertClassificationTask.Config, NewBertPairClassificationTask.Config, BertPairRegressionTask.Config, WordTaggingTask.Config, IntentSlotTask.Config, LMTask.Config, MaskedLMTask.Config, PairwiseClassificationTask.Config, PairwiseRegressionTask.Config, PairwiseClassificationForDenseRetrievalTask.Config, RoBERTaNERTask.Config, SeqNNTask.Config, SquadQATask.Config, SemanticParsingTask.Config, SequenceLabelingTask.Config]

use_cuda_if_available: bool = True

use_fp16: bool = False

distributed_world_size: int = 1

gpu_streams_for_distributed_training: int = 1

load_snapshot_path: str = ''

save_snapshot_path: str = '/tmp/model.pt'

use_config_from_snapshot: bool = True

auto_resume_from_snapshot: bool = False

export: ExportConfig = ExportConfig()

export_list: list[ExportConfig] = []

modules_save_dir: str = ''

save_module_checkpoints: bool = False

save_all_checkpoints: bool = False

use_tensorboard: bool = True

random_seed: Optional[int] = 0

Seed value to seed torch, python, and numpy random generators.

use_deterministic_cudnn: bool = False

Whether to allow CuDNN to behave deterministically.

report_eval_results: bool = False

report_test_results: bool = True

include_dirs: Optional[list[str]] = None

version: int

use_cuda_for_testing: bool = True

read_chunk_size: Optional[int] = 1000000000

test_out_path: str = '/tmp/test_out.txt'

debug_path: str = '/tmp/model.debug'

Warning

This config has parameters with no default values. We aren’t yet able to generate functional JSON for it.