PyTorch Lightning wrapper for LayoutLM family of models.
- Install
poetry
- Install dependencies using
poetry install
$ python scripts/train.py --data_dir <data_path>
- MLFLow / Logging integration
- Add logger
- Log training metrics
- Log validation metrics per epoch
- Validation Loop
- Generate validation predictions per step as IOB
- Generate validation predictions per epoch
- Inference, torchscript, quantization
- LayoutLMv2
- LayoutLMv3