Skip to main content

Module optimizer

Module optimizer 

Source
Expand description

Burn optimizer wrapper used by both PPO and DQN. Burn optimizer wrapper used by the PPO and DQN trainers.

After phase 5 of the Burn migration (#82), Burn is the only tensor backend in the workspace; the backend-agnostic abstraction added in phase 2b (#92) has collapsed to a single Burn impl. The BackendOptimizer trait survives as a minimal interface so the trainer bodies can hold the optimizer behind a generic without naming the concrete Burn Optimizer<M, B> type.

Structs§

BurnOptimizer
Burn-side optimizer wrapper.

Traits§

BackendOptimizer
Burn-side optimizer interface used by PPO and DQN trainers.

Functions§

wrap_burn
Helper: wrap a freshly-built Burn optimizer in a BurnOptimizer.

Type Aliases§

OptimResult
Result alias used by trainer-side optimizer plumbing.