rxn.onmt_utils.model_introspection.get_model_rnn_size

rxn.onmt_utils.model_introspection.get_model_rnn_size(model_path)[source]

Get the value of rnn_size for the given model checkpoint.

Parameters

model_path (Union[str, PathLike]) – model checkpoint, such as model_step_100000.pt.

Return type

int