Change checkpoint filename without overwriting full filepath

[wietsedv]

🚀 Feature

By default, the checkpoint path resolves to lightning_logs/{version}/checkpoints/{epoch}.ckpt. It is currently not possible to change the checkpoint filename without replacing the full path.

Motivation

The full path is replacable with the ModelCheckpoint.filepath argument. This feature is undisirable because I want to use the default path resolver that uses the options from Trainer instead of hardcoding a path.

A satisfying path construction workaraound based on values in Trainer is not possible because the Trainer is constructed after the ModelCheckpoint.

Pitch

I would like to add a filename parameter to the ModelCheckpoint constructor. This parameter is complementary to filepath and filepath will keep working like it does for backward compatibility.

• If only filepath is provided, the original behaviour is used.
• If both filepath and filename are provided, the checkpoint path will resolve to filepath/filename.
• If only filename is provided, the checkpoint path will resolve to resolved directory/filename (instead of resolved directory/epoch).

[stale]

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.