Add gc_after_trial flag to avoid memory issues.

HumanCompatibleAI · Feb 5, 2024 · b2a98d7 · b2a98d7
1 parent 147c996
commit b2a98d7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tuning/tune.py b/tuning/tune.py
@@ -77,7 +77,8 @@ def main():
  run_options={"--name": study.study_name, "--file_storage": "sacred"},
  extra_named_configs=args.named_configs
  ),
- callbacks=[optuna.study.MaxTrialsCallback(args.num_trials)]
+ callbacks=[optuna.study.MaxTrialsCallback(args.num_trials)],
+ gc_after_trial=True,
  )