Skip to content

Add option to replace bn by sync bn in auto_model #1261

@vfdev-5

Description

@vfdev-5

🚀 Feature

Idea is to add an option sync_bn (default, False) to apply convert_sync_batchnorm on the input model if a DDP wrapper is used further.

  • Check performances on PascalVOC
  • Update ref examples

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions