_NewTask.ConfigΒΆ

Component: _NewTask

class _NewTask.Config[source]

Bases: ConfigBase

All Attributes (including base classes)

data: Data.Config = Data.Config()
trainer: TaskTrainer.Config = TaskTrainer.Config()
use_elastic: Optional[bool] = None
Subclasses
  • NewDisjointMultitask.Config
  • NewTask.Config
  • BertPairRegressionTask.Config
  • DocumentClassificationTask.Config
  • DocumentRegressionTask.Config
  • EnsembleTask.Config
  • IntentSlotTask.Config
  • LMTask.Config
  • MaskedLMTask.Config
  • NewBertClassificationTask.Config
  • NewBertPairClassificationTask.Config
  • PairwiseClassificationForDenseRetrievalTask.Config
  • PairwiseClassificationTask.Config
  • PairwiseRegressionTask.Config
  • QueryDocumentPairwiseRankingTask.Config
  • RoBERTaNERTask.Config
  • SemanticParsingTask.Config
  • SeqNNTask.Config
  • SequenceLabelingTask.Config
  • SquadQATask.Config
  • WordTaggingTask.Config

Default JSON

{
    "data": {
        "Data": {
            "source": {
                "TSVDataSource": {
                    "column_mapping": {},
                    "train_filename": null,
                    "test_filename": null,
                    "eval_filename": null,
                    "field_names": null,
                    "delimiter": "\t",
                    "quoted": false,
                    "drop_incomplete_rows": false
                }
            },
            "batcher": {
                "PoolingBatcher": {
                    "train_batch_size": 16,
                    "eval_batch_size": 16,
                    "test_batch_size": 16,
                    "pool_num_batches": 1000,
                    "num_shuffled_pools": 1
                }
            },
            "sort_key": null,
            "in_memory": true
        }
    },
    "trainer": {
        "TaskTrainer": {
            "epochs": 10,
            "early_stop_after": 0,
            "max_clip_norm": null,
            "report_train_metrics": true,
            "target_time_limit_seconds": null,
            "do_eval": true,
            "load_best_model_after_train": true,
            "num_samples_to_log_progress": 1000,
            "num_accumulated_batches": 1,
            "num_batches_per_epoch": null,
            "optimizer": {
                "Adam": {
                    "lr": 0.001,
                    "weight_decay": 1e-05,
                    "eps": 1e-08
                }
            },
            "scheduler": null,
            "sparsifier": null,
            "fp16_args": {
                "FP16OptimizerFairseq": {
                    "init_loss_scale": 128,
                    "scale_window": null,
                    "scale_tolerance": 0.0,
                    "threshold_loss_scale": null,
                    "min_loss_scale": 0.0001
                }
            },
            "privacy_engine": null,
            "use_tensorboard": false
        }
    },
    "use_elastic": null
}