Skip to content

Add InstanceNorm3dNVFuser based on NVIDIA Apex #4085

@Nic-Ma

Description

@Nic-Ma

Is your feature request related to a problem? Please describe.
Fast InstanceNorm was added via NVFuser in APEX and can be used as a drop-in replacement for nn.InstanceNorm3d.
Would be great to have a way to pass additional parameters to the norm layer, since it might be useful when fusions are added.

Will implement InstanceNorm3dNVFuser, starting from DynUNet.

Metadata

Metadata

Type

No type

Projects

Relationships

None yet

Development

No branches or pull requests

Issue actions