rxn.onmt_utils.train_command

Functions

preprocessed_id_names(n_additional_sets)

Get the names of the ids for the datasets used in multi-task training with OpenNMT.

Classes

Arg(key, default, needed_for)

Represents an argument to be given for the onmt_train command.

OnmtTrainCommand(command_type, no_gpu, ...)

Class to build the onmt_command for training models, continuing the training, or finetuning.

RxnCommand(value)

Flag indicating which command(s) the parameters relate to.