mirror of
https://github.com/explosion/spaCy.git
synced 2024-12-25 01:16:28 +03:00
ff84075839
* Support infinite generators for training corpora Support a training corpus with an infinite generator in the `spacy train` training loop: * Revert `create_train_batches` to the state where an infinite generator can be used as the in the first epoch of exactly one epoch without resulting in a memory leak (`max_epochs != 1` will still result in a memory leak) * Move the shuffling for the first epoch into the corpus reader, renaming it to `spacy.Corpus.v2`. * Switch to training option for shuffling in memory Training loop: * Add option `training.shuffle_train_corpus_in_memory` that controls whether the corpus is loaded in memory once and shuffled in the training loop * Revert changes to `create_train_batches` and rename to `create_train_batches_with_shuffling` for use with `spacy.Corpus.v1` and a corpus that should be loaded in memory * Add `create_train_batches_without_shuffling` for a corpus that should not be shuffled in the training loop: the corpus is merely batched during training Corpus readers: * Restore `spacy.Corpus.v1` * Add `spacy.ShuffledCorpus.v1` for a corpus shuffled in memory in the reader instead of the training loop * In combination with `shuffle_train_corpus_in_memory = False`, each epoch could result in a different augmentation * Refactor create_train_batches, validation * Rename config setting to `training.shuffle_train_corpus` * Refactor to use a single `create_train_batches` method with a `shuffle` option * Only validate `get_examples` in initialize step if: * labels are required * labels are not provided * Switch back to max_epochs=-1 for streaming train corpus * Use first 100 examples for stream train corpus init * Always check validate_get_examples in initialize
132 lines
3.9 KiB
INI
132 lines
3.9 KiB
INI
[paths]
|
|
train = null
|
|
dev = null
|
|
vectors = null
|
|
init_tok2vec = null
|
|
|
|
[system]
|
|
seed = 0
|
|
gpu_allocator = null
|
|
|
|
[nlp]
|
|
lang = null
|
|
# List of pipeline component names, in order. The names should correspond to
|
|
# components defined in the [components block]
|
|
pipeline = []
|
|
# Components that are loaded but disabled by default
|
|
disabled = []
|
|
# Optional callbacks to modify the nlp object before it's initialized, after
|
|
# it's created and after the pipeline has been set up
|
|
before_creation = null
|
|
after_creation = null
|
|
after_pipeline_creation = null
|
|
# Default batch size to use with nlp.pipe and nlp.evaluate
|
|
batch_size = 1000
|
|
|
|
[nlp.tokenizer]
|
|
@tokenizers = "spacy.Tokenizer.v1"
|
|
|
|
# The pipeline components and their models
|
|
[components]
|
|
|
|
# Readers for corpora like dev and train.
|
|
[corpora]
|
|
|
|
[corpora.train]
|
|
@readers = "spacy.Corpus.v1"
|
|
path = ${paths.train}
|
|
# Whether to train on sequences with 'gold standard' sentence boundaries
|
|
# and tokens. If you set this to true, take care to ensure your run-time
|
|
# data is passed in sentence-by-sentence via some prior preprocessing.
|
|
gold_preproc = false
|
|
# Limitations on training document length
|
|
max_length = 0
|
|
# Limitation on number of training examples
|
|
limit = 0
|
|
# Apply some simply data augmentation, where we replace tokens with variations.
|
|
# This is especially useful for punctuation and case replacement, to help
|
|
# generalize beyond corpora that don't/only have smart quotes etc.
|
|
augmenter = null
|
|
|
|
[corpora.dev]
|
|
@readers = "spacy.Corpus.v1"
|
|
path = ${paths.dev}
|
|
# Whether to train on sequences with 'gold standard' sentence boundaries
|
|
# and tokens. If you set this to true, take care to ensure your run-time
|
|
# data is passed in sentence-by-sentence via some prior preprocessing.
|
|
gold_preproc = false
|
|
# Limitations on training document length
|
|
max_length = 0
|
|
# Limitation on number of training examples
|
|
limit = 0
|
|
# Optional callback for data augmentation
|
|
augmenter = null
|
|
|
|
# Training hyper-parameters and additional features.
|
|
[training]
|
|
seed = ${system.seed}
|
|
gpu_allocator = ${system.gpu_allocator}
|
|
dropout = 0.1
|
|
accumulate_gradient = 1
|
|
# Controls early-stopping. 0 disables early stopping.
|
|
patience = 1600
|
|
# Number of epochs. 0 means unlimited. If >= 0, train corpus is loaded once in
|
|
# memory and shuffled within the training loop. -1 means stream train corpus
|
|
# rather than loading in memory with no shuffling within the training loop.
|
|
max_epochs = 0
|
|
max_steps = 20000
|
|
eval_frequency = 200
|
|
# Control how scores are printed and checkpoints are evaluated.
|
|
score_weights = {}
|
|
# Names of pipeline components that shouldn't be updated during training
|
|
frozen_components = []
|
|
# Location in the config where the dev corpus is defined
|
|
dev_corpus = "corpora.dev"
|
|
# Location in the config where the train corpus is defined
|
|
train_corpus = "corpora.train"
|
|
# Optional callback before nlp object is saved to disk after training
|
|
before_to_disk = null
|
|
|
|
[training.logger]
|
|
@loggers = "spacy.ConsoleLogger.v1"
|
|
|
|
[training.batcher]
|
|
@batchers = "spacy.batch_by_words.v1"
|
|
discard_oversize = false
|
|
tolerance = 0.2
|
|
|
|
[training.batcher.size]
|
|
@schedules = "compounding.v1"
|
|
start = 100
|
|
stop = 1000
|
|
compound = 1.001
|
|
|
|
[training.optimizer]
|
|
@optimizers = "Adam.v1"
|
|
beta1 = 0.9
|
|
beta2 = 0.999
|
|
L2_is_weight_decay = true
|
|
L2 = 0.01
|
|
grad_clip = 1.0
|
|
use_averages = false
|
|
eps = 1e-8
|
|
learn_rate = 0.001
|
|
|
|
# These settings are used when nlp.initialize() is called (typically before
|
|
# training or pretraining). Components and the tokenizer can each define their
|
|
# own arguments via their initialize methods that are populated by the config.
|
|
# This lets them gather data resources, build label sets etc.
|
|
[initialize]
|
|
vectors = ${paths.vectors}
|
|
# Extra resources for transfer-learning or pseudo-rehearsal
|
|
init_tok2vec = ${paths.init_tok2vec}
|
|
# Data and lookups for vocabulary
|
|
vocab_data = null
|
|
lookups = null
|
|
# Arguments passed to the tokenizer's initialize method
|
|
tokenizer = {}
|
|
# Arguments for initialize methods of the components (keyed by component)
|
|
components = {}
|
|
before_init = null
|
|
after_init = null
|