-
Notifications
You must be signed in to change notification settings - Fork 705
Open
Description
lightning-hydra-template/src/models/mnist_module.py
Lines 202 to 213 in bddbc24
if self.hparams.scheduler is not None: | |
scheduler = self.hparams.scheduler(optimizer=optimizer) | |
return { | |
"optimizer": optimizer, | |
"lr_scheduler": { | |
"scheduler": scheduler, | |
"monitor": "val/loss", | |
"interval": "epoch", | |
"frequency": 1, | |
}, | |
} | |
return {"optimizer": optimizer} |
There is an issue with not being able to change the monitor interval frequency value of the lr_scheduler using the config.
If I need to change the learning rate at each step, I will have to modify the module file directly.
Metadata
Metadata
Assignees
Labels
No labels