espnet3.parallel.parallel.DictReturnWorkerPlugin

source

class espnet3.parallel.parallel.DictReturnWorkerPlugin(setup_fn: Callable[[], dict])

Bases: WorkerPlugin

A WorkerPlugin that calls a user-defined setup function once per worker, and stores the returned dictionary in worker.plugins[“env”].

setup(worker)

Run when the plugin is attached to a worker. This happens when the plugin is registered and attached to existing workers, or when a worker is created after the plugin has been registered.