# Trainer

## Introduction

Trainer from [lightning](https://pytorch-lightning.readthedocs.io/en/stable/common/trainer.html), with support cross-validation feature.

## Usage

### Cross-validation

Set `num_folds` argument to an int bigger than `1`, to use the cross-validation feature with `num_folds` folds.