meerqat.train.save_ptm module#
usage: save_ptm.py [-h] [–config <config>] [–bert] <ckpt>
Save the PreTrainedModel(s) wrapped inside the Trainee (LightningModule).
- positional arguments:
<ckpt> Path to the lightning checkpoint.
- options:
- -h, --help
show this help message and exit
- --config <config>
Path to the lightning config file (YAML).
- --bert
For DPR-based BiEncoder, save BertModel instead of DPR*Encoder