Pytorch Lightning ModelCheckpoint не сохраняет последнюю контрольную точку проверки, когда save_last = True ⇐ Python

[Anonymous]

Я использую pytorch Lightning для обучения своей модели. Здесь я использую обратный вызов Lightning ModelCheckpoint со следующими настройками:

Код: Выделить всё

ModelCheckpoint(
dirpath="path/to/dir",
monitor="val_loss",
mode="min",
filename="best_model",
save_top_k=1,
save_last=True,
)

Затем я провожу обучение с помощью:

Код: Выделить всё

trainer = Trainer(
callbacks = [model_checkpoint],
val_check_interval = 0.2,
max_epochs = 20,
accelerator = "gpu",
log_every_n_steps = 1.
)

trainer.fit(
model=model,
datamodule=data_module,
)

Я ожидаю, что эти настройки будут сохранены в файлах контрольных точек: best_model.ckpt, который имеет контрольную точку модели с лучшим показателем val_loss, и last.ckpt, который будет контрольной точкой последней проверенной модели.
Я заметил, что параметр save_last работал не так, как я ожидал, вместо этого: "Когда True, сохраняет копию last.ckpt при каждом сохранении файла контрольной точки». - Документация по молниям. Это означает, что если для save_top_k установлено значение 1, best_model.ckpt и last.ckpt будут абсолютно одинаковыми.
Тогда у меня такие вопросы: Как я могу сохранить как контрольную точку модели, показавшую наилучшие результаты при проверке, так и последнюю проверенную модель?

Подробнее здесь: https://stackoverflow.com/questions/798 ... nt-when-sa