rxn.onmt_utils.strip_model.strip_model

rxn.onmt_utils.strip_model.strip_model(model_in, model_out)[source]

Strip the optim data of the given OpenNMT model.

This usually reduces the size of the file by 2/3.

Parameters

  • model_in (Path) – path of the model to strip.

  • model_out (Path) – where to save the new model (can be identical to model_in).

Return type

None