Skip to content
Snippets Groups Projects

Train

Use the teklia-dan train command to train a new DAN model.

Two subcommands are available depending on your dataset:

line
Train a DAN model at line-level.
document
Train a DAN model at document-level.

Remarks (for pre-training and training)

All hyperparameters are specified and editable in the training scripts (meaning are in comments).

Evaluation is performed just after training ending (training is stopped when the maximum elapsed time is reached or after a maximum number of epoch as specified in the training script).

The outputs files are split into two subfolders:

checkpoints
Contains model weights for the last trained epoch and for the epoch giving the best valid CER.
results
Contains tensorboard log for loss and metrics as well as text file for used hyperparameters and results of evaluation.