Hi there!
I got an error about an local variable referenced before assignment:
Traceback (most recent call last):
  File "/home/skrix/.conda/envs/new_deeplink_env/lib/python3.8/runpy.py", line 194, in _run_module_as_main
    return _run_code(code, main_globals, None,
  File "/home/skrix/.conda/envs/new_deeplink_env/lib/python3.8/runpy.py", line 87, in _run_code
    exec(code, run_globals)
  File "/home/skrix/mavo/src/deeplink/__main__.py", line 8, in <module>
    main()
  File "/home/skrix/.conda/envs/new_deeplink_env/lib/python3.8/site-packages/click/core.py", line 829, in __call__
    return self.main(*args, **kwargs)
  File "/home/skrix/.conda/envs/new_deeplink_env/lib/python3.8/site-packages/click/core.py", line 782, in main
    rv = self.invoke(ctx)
  File "/home/skrix/.conda/envs/new_deeplink_env/lib/python3.8/site-packages/click/core.py", line 1259, in invoke
    return _process_result(sub_ctx.command.invoke(sub_ctx))
  File "/home/skrix/.conda/envs/new_deeplink_env/lib/python3.8/site-packages/click/core.py", line 1259, in invoke
    return _process_result(sub_ctx.command.invoke(sub_ctx))
  File "/home/skrix/.conda/envs/new_deeplink_env/lib/python3.8/site-packages/click/core.py", line 1066, in invoke
    return ctx.invoke(self.callback, **ctx.params)
  File "/home/skrix/.conda/envs/new_deeplink_env/lib/python3.8/site-packages/click/core.py", line 610, in invoke
    return callback(*args, **kwargs)
  File "/home/skrix/mavo/src/deeplink/cli.py", line 785, in run_opt
    run_bayesian_hyperopt(
  File "/home/skrix/mavo/src/deeplink/rgcn_hetero/deployer.py", line 410, in run_bayesian_hyperopt
    study.optimize(func=objective, n_trials=n_trials, timeout=timeout)
  File "/home/skrix/.conda/envs/new_deeplink_env/lib/python3.8/site-packages/optuna/study.py", line 385, in optimize
    _optimize(
  File "/home/skrix/.conda/envs/new_deeplink_env/lib/python3.8/site-packages/optuna/_optimize.py", line 66, in _optimize
    _optimize_sequential(
  File "/home/skrix/.conda/envs/new_deeplink_env/lib/python3.8/site-packages/optuna/_optimize.py", line 163, in _optimize_sequential
    trial = _run_trial(study, func, catch)
  File "/home/skrix/.conda/envs/new_deeplink_env/lib/python3.8/site-packages/optuna/_optimize.py", line 268, in _run_trial
    raise func_err
  File "/home/skrix/.conda/envs/new_deeplink_env/lib/python3.8/site-packages/optuna/_optimize.py", line 217, in _run_trial
    value_or_values = func(trial)
  File "/home/skrix/mavo/src/deeplink/rgcn_hetero/deployer.py", line 370, in objective
    trainer.fit(
  File "/home/skrix/.conda/envs/new_deeplink_env/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 499, in fit
    self.dispatch()
  File "/home/skrix/.conda/envs/new_deeplink_env/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 546, in dispatch
    self.accelerator.start_training(self)
  File "/home/skrix/.conda/envs/new_deeplink_env/lib/python3.8/site-packages/pytorch_lightning/accelerators/accelerator.py", line 73, in start_training
    self.training_type_plugin.start_training(trainer)
  File "/home/skrix/.conda/envs/new_deeplink_env/lib/python3.8/site-packages/pytorch_lightning/plugins/training_type/training_type_plugin.py", line 114, in start_training
    self._results = trainer.run_train()
  File "/home/skrix/.conda/envs/new_deeplink_env/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 637, in run_train
    self.train_loop.run_training_epoch()
  File "/home/skrix/.conda/envs/new_deeplink_env/lib/python3.8/site-packages/pytorch_lightning/trainer/training_loop.py", line 492, in run_training_epoch
    batch_output = self.run_training_batch(batch, batch_idx, dataloader_idx)
  File "/home/skrix/.conda/envs/new_deeplink_env/lib/python3.8/site-packages/pytorch_lightning/trainer/training_loop.py", line 654, in run_training_batch
    self.optimizer_step(optimizer, opt_idx, batch_idx, train_step_and_backward_closure)
  File "/home/skrix/.conda/envs/new_deeplink_env/lib/python3.8/site-packages/pytorch_lightning/trainer/training_loop.py", line 425, in optimizer_step
    model_ref.optimizer_step(
  File "/home/skrix/.conda/envs/new_deeplink_env/lib/python3.8/site-packages/pytorch_lightning/core/lightning.py", line 1390, in optimizer_step
    optimizer.step(closure=optimizer_closure)
  File "/home/skrix/.conda/envs/new_deeplink_env/lib/python3.8/site-packages/pytorch_lightning/core/optimizer.py", line 214, in step
    self.__optimizer_step(*args, closure=closure, profiler_name=profiler_name, **kwargs)
  File "/home/skrix/.conda/envs/new_deeplink_env/lib/python3.8/site-packages/pytorch_lightning/core/optimizer.py", line 134, in __optimizer_step
    trainer.accelerator.optimizer_step(optimizer, self._optimizer_idx, lambda_closure=closure, **kwargs)
  File "/home/skrix/.conda/envs/new_deeplink_env/lib/python3.8/site-packages/pytorch_lightning/accelerators/accelerator.py", line 277, in optimizer_step
    self.run_optimizer_step(optimizer, opt_idx, lambda_closure, **kwargs)
  File "/home/skrix/.conda/envs/new_deeplink_env/lib/python3.8/site-packages/pytorch_lightning/accelerators/accelerator.py", line 282, in run_optimizer_step
    self.training_type_plugin.optimizer_step(optimizer, lambda_closure=lambda_closure, **kwargs)
  File "/home/skrix/.conda/envs/new_deeplink_env/lib/python3.8/site-packages/pytorch_lightning/plugins/training_type/training_type_plugin.py", line 163, in optimizer_step
    optimizer.step(closure=lambda_closure, **kwargs)
  File "/home/skrix/.conda/envs/new_deeplink_env/lib/python3.8/site-packages/torch/optim/optimizer.py", line 89, in wrapper
    return func(*args, **kwargs)
  File "/home/skrix/.conda/envs/new_deeplink_env/lib/python3.8/site-packages/torch/autograd/grad_mode.py", line 27, in decorate_context
    return func(*args, **kwargs)
  File "/home/skrix/.conda/envs/new_deeplink_env/lib/python3.8/site-packages/torch/optim/adamw.py", line 117, in step
    beta1,
UnboundLocalError: local variable 'beta1' referenced before assignment

``

Do you have an idea why? 

Pytorch Version: 1.8.0