Получите пути к сохраненным контрольным точкам из Pytorch-Lightning ModelCheckpoint. ⇐ Python

[Anonymous]

Я использую PytorchLightning и ModelCheckpoint, который сохраняет модели с отформатированным именем файла, например filename="model_{epoch}-{val_acc:.2f}"
Позже я хочу снова получить доступ к этим контрольным точкам. Для простоты я хочу получить максимум от save_top_k=N.
Поскольку имя файла является динамическим, мне интересно, как легко получить файлы контрольных точек.

Есть ли встроенный атрибут в ModelCheckpoint или в трейнере, который дает мне сохраненные контрольные точки?
Например, например

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

checkpoint_callback.get_top_k_paths()

Я знаю, что могу сделать это с помощью glob и model_dir. Я предполагаю, что обратный вызов в любом случае должен их отслеживать, поэтому мне интересно, какой встроенный способ доступа к путям модели.

Подробнее здесь: https://stackoverflow.com/questions/745 ... checkpoint