rxn.onmt_utils.model_resize.ModelResizer
- class rxn.onmt_utils.model_resize.ModelResizer(model_path)[source]
Bases:
object- Parameters
model_path (
Union[str,PathLike]) –
- __init__(model_path)[source]
Resizing pretrained onmt models for training on extended vocab.
- Parameters
model_path (
Union[str,PathLike]) – Path to model checkpoint
Methods
__init__(model_path)Resizing pretrained onmt models for training on extended vocab.
extend_vocab(new_vocab_path)Extend vocab and size of a model using a new vocab file s :type new_vocab_path:
Union[str,PathLike] :param new_vocab_path: Path to new vocab file (vocab.pt generated by ONMT)save_checkpoint(save_path)Save checkpoint of resized model