Skip to main content
Version: v2602

Auto Pruning

ZenithTune provides auto pruners that automatically prune trials during tuning. This feature helps avoid excessively time-consuming executions, efficiently utilize resources, and reduce tuning time by early termination of unpromising trials.

Pruner Types​

TimeoutPruner​

Prunes trials that exceed the specified time limit.

from aibooster.intelligence.zenith_tune.auto_pruners import TimeoutPruner
from aibooster.intelligence.zenith_tune import CommandOutputTuner

# Pruner that times out after 300 seconds
timeout_pruner = TimeoutPruner(timeout_seconds=300.0)

tuner = CommandOutputTuner(
    auto_pruners=[timeout_pruner]
)

Usage Examples​

Please refer to the following example file:

  • <AIBooster repository>/aibooster/examples/intelligence/zenith_tune/pruners/timeout_pruner_example.py