Skip to content
Snippets Groups Projects
Commit 729a7e9b authored by Daniel CARRON's avatar Daniel CARRON :b:
Browse files

Renamed optimizer_params to optimizer_configs

parent c5f71ad4
No related branches found
No related tags found
1 merge request!4Moved code to lightning
......@@ -114,7 +114,7 @@ class Densenet(pl.LightningModule):
def configure_optimizers(self):
# Dynamically instantiates the optimizer given the configs
optimizer = getattr(torch.optim, self.hparams.optimizer)(
self.parameters(), **self.hparams.optimizer_params
self.parameters(), **self.hparams.optimizer_configs
)
return optimizer
......@@ -205,7 +205,7 @@ class PASA(pl.LightningModule):
def configure_optimizers(self):
# Dynamically instantiates the optimizer given the configs
optimizer = getattr(torch.optim, self.hparams.optimizer)(
self.parameters(), **self.hparams.optimizer_params
self.parameters(), **self.hparams.optimizer_configs
)
return optimizer
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment