Skip to content

Enhance some networks to set bias=False for conv layers #2768

@yiheng-wang-nv

Description

@yiheng-wang-nv

If a conv layer is directly followed by a batch norm layer, the bias in the conv layer is not need, because the subtract the mean manipulation will cancel out the effect of bias.

Reference:

In MONAI's Convolution class, the default value of bias is True, and the following networks/blocks cannot configure the bias parameter when using Convolution:

  • SimpleASPP
  • HighResNet
  • VNet
  • UNet
  • BasicUNet
  • AutoEncoder

Metadata

Metadata

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions