trainΒΆ

Training or finetuning a LSR model on DocRED dataset.

Functions

get_optimizer

lsr_collate_fn

Manually processes labels and uses default for the rest.

parse_args

train_model

Classes

DocredDataset

LsrMetrics

For calculating metrics for LsrModel.

MyAdagrad

Modification of the Adagrad optimizer that allows to specify an initial accumulator value.