Skip to content

Commit

Permalink
[brief] Fixes the optuna example.
Browse files Browse the repository at this point in the history
[detailed]
- Forgot to change the index.
  • Loading branch information
marovira committed Aug 26, 2024
1 parent e4e36ca commit 79c5bd7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/optuna_tutorial.py
Original file line number Diff line number Diff line change
Expand Up @@ -111,7 +111,7 @@ def metrics(self) -> dict[str, typing.Any]:
return {"accuracy": accuracy}

def setup(self, fast_init: bool = False) -> None:
plugin = self.trainer.plugins[0]
plugin = self.trainer.plugins["optuna"]
assert isinstance(plugin, OptunaPlugin)

# Assign the tunable parameters so we can log them as hyper-parameters when
Expand Down

0 comments on commit 79c5bd7

Please sign in to comment.