Skip to content

Commit

Permalink
Create optimizer with method in application (#114)
Browse files Browse the repository at this point in the history
  • Loading branch information
BenjaminMidtvedt authored Jun 25, 2024
1 parent 2aa09ce commit 85a51bb
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion deeplay/applications/application.py
Original file line number Diff line number Diff line change
Expand Up @@ -232,7 +232,7 @@ def compute_loss(self, y_hat, y) -> Union[torch.Tensor, Dict[str, torch.Tensor]]
def configure_optimizers(self):
try:

return self.optimizer.create()
return self.create_optimizer_with_params(self.optimizer, self.parameters())

except AttributeError as e:
raise AttributeError(
Expand Down

0 comments on commit 85a51bb

Please sign in to comment.