DistSamplerSeedHook — mmengine 0.10.7 documentation (original) (raw)

• Docs >
• mmengine.hooks >
• DistSamplerSeedHook
• 以中文阅读

Shortcuts

class mmengine.hooks.DistSamplerSeedHook[source]¶

Data-loading sampler for distributed training.

When distributed training, it is only useful in conjunction withEpochBasedRunner, while IterBasedRunner achieves the same purpose with IterLoader.

before_train_epoch(runner)[source]¶

Set the seed for sampler and batch_sampler.

Parameters:

runner (Runner) – The runner of the training process.

Return type:

None

• DistSamplerSeedHook
  • DistSamplerSeedHook
    * DistSamplerSeedHook.before_train_epoch()