trw.train.options

Module Contents

Functions

create_default_options(logging_directory=None, num_epochs=50, device=None)

Create default options for the training and evaluation process.
trw.train.options.create_default_options(logging_directory=None, num_epochs=50, device=None)

Create default options for the training and evaluation process.

Parameters

  • logging_directory – the base directory where the logs will be exported for each trained model. If None and if the environment variable LOGGING_DIRECTORY exists, it will be used as root directory. Else a default folder will be used

  • num_epochs – the number of epochs

  • device – the device to train the model on. If None, we will try first any available GPU then revert to CPU

Returns

the options