Skip to content

Commit

Permalink
Update comment
Browse files Browse the repository at this point in the history
  • Loading branch information
mepland committed Jan 16, 2024
1 parent 3cbdd32 commit 0436c7d
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion utils/bayesian_opt.py
Original file line number Diff line number Diff line change
Expand Up @@ -215,7 +215,8 @@ def complete_iter(
print(f"\nStarting {i_iter = }, with {n_points = }")
next_point_to_probe = optimizer.suggest(utility)

# Create a fresh model_wrapper object to try to avoid GPU memory leaks TODO probably can safely revert
# Create a fresh model_wrapper object to try to avoid GPU memory leaks
# This may not be necessary, but as it is already coded, just be safe and leave it
model_wrapper = model_wrapper_class(TSModelWrapper=parent_wrapper)
model_wrapper.set_work_dir(work_dir_absolute=bayesian_opt_work_dir)
model_wrapper.set_enable_progress_bar_and_max_time(
Expand Down

0 comments on commit 0436c7d

Please sign in to comment.