исправить количество контрольных точек модели, которые будут сохранены pytorch

Question

Есть ли что-нибудь похожее на max_to_keep в tf.Saver () в pytorch? Как я могу сохранить только количество контрольных точек модели? Есть ли для этого встроенная функция горелки?

Я использую функцию ниже для сохранения контрольных точек.

import os
import torch

def save_checkpoint(model, save_path):
    if not os.path.exists(os.path.dirname(save_path)):
        os.makedirs(os.path.dirname(save_path))

    torch.save(model.cpu().state_dict(), save_path)
    model.cuda()