train

Primary training and evaluation script. Run python3 train.py -h to see available options.

Module Contents

train.main(hparams)

Main training routine specific for this project

Parameters

hparams (Namespace) – Namespace containing configuration values

train.set_hparams(hparams)

Add constant parameter values based on passed arguments.

Parameters

hparams (Namespace) – Parameters

Returns

Modified parameters

Return type

Namespace

train.get_model(hparams)

Prepare model and the data.

Parameters

hparams (Namespace) – Holds configuration values.

Raises

ImportError – The requested model and prediction data must be compatible.

Returns

Model with the linked data.

Return type

Model