Skip to content

Custom Checkpoint file extension #4963

@Borda

Description

@Borda

🚀 Feature

Atm, we have hardcoded .ckpt as a file extension for any checkpoint.

https://github.com/PyTorchLightning/pytorch-lightning/blob/db69d169e868451c16a4ee418f2de835ccce146c/pytorch_lightning/callbacks/model_checkpoint.py#L429

Proposed solution 1:

class ModelCheckpoint(Callback):
    FILE_EXTENSION = ".ckpt"

Proposed solution 2:

ModelCheckpoint(ext=''pt')

Metadata

Metadata

Labels

checkpointingRelated to checkpointingfeatureIs an improvement or enhancementgood first issueGood for newcomers

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions