intelligence.zenith_tune.presets.demo

Demo preset for fake_train.py tuning.

DemoPreset Objects

@PresetRegistry.register("dev:demo")
class DemoPreset(TuningPreset)

Preset for tuning fake_train.py parameters.

Optimizes --batch-size, --num-workers (command-line args) and OMP_NUM_THREADS (environment variable).

Example:

zenithtune optimize --preset dev:demo --dev -- python fake_train.py --epochs 3