Как мне зарегистрировать гиперпараметр в Lightning? ⇐ Python

[Anonymous]

как я могу зарегистрировать гиперпараметр, который представляет собой строку в модуле Lightning? Я хочу использовать эту hpram для сохранения моей модели:

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

def experiment_name(exp):
ckp = ModelCheckpoint(
dirpath=f"/content/aims_proj/saved_models/{exp}",
filename="model-{epoch:02d}-{val_loss:.2f}-{criterion}", #criterion is a strign hparam
save_top_k=1,
verbose=False,
monitor="val_loss",
mode="min"
)

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

class BaseModel(L.LightningModule):

def __init__(self, criterion, lr, max_iters, warmup):
super().__init__()
self.criterion = Loss(criterion)
self.save_hyperparameters()
self.log(f"criterion", criterion)

Я попробовал, как показано в блоке кода выше. Я просто хочу, чтобы сохраненная модель включала гиперпаметр.


Подробнее здесь: https://stackoverflow.com/questions/787 ... -lightning