Skip to content

Commit

Permalink
fix README of normalizePlugin
Browse files Browse the repository at this point in the history
  • Loading branch information
lcxywfe authored and rajeevsrao committed Mar 15, 2020
1 parent b447156 commit 8cca9a8
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions plugin/normalizePlugin/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ This plugin consists of the plugin creator class `NormalizePluginCreator` and th
|`bool` |`acrossSpatial` |When `acrossSpatial = true`, the normalization is conducted for each sample from the batch independently, meaning across spatial locations only. This is somewhat similar to instance normalization. When `acrossSpatial = false`, the normalization is conducted across all samples in the batch and spatial locations. This is somewhat similar to batch normalization.
|`bool` |`channelShared` |When `channelShared = true`, a single scale factor is used for all channels. When `channelShared = false`, several scale factors are provided and are used for each channel independently.
|`float` |`eps` |A small number to prevent being divided by zero during normalization.
|`Weights *` |`weights` |A pointer to weights which contains information about scale factors for normalization. The definition of `Weights` can be found in the [NvInfer.h](https://docs.nvidia.com/deeplearning/sdk/tensorrt-api/c_api/_nv_infer_8h_source.html) header. The number of values in weights is `1` if `channelShared = false`, otherwise the number of values in weights is the number of channels in the input tensor.
|`Weights *` |`weights` |A pointer to weights which contains information about scale factors for normalization. The definition of `Weights` can be found in the [NvInfer.h](https://docs.nvidia.com/deeplearning/sdk/tensorrt-api/c_api/_nv_infer_8h_source.html) header. The number of values in weights is `1` if `channelShared = true`, otherwise the number of values in weights is the number of channels in the input tensor.
|`int` |`nbWeights` |The number of weights sent to the plugin. This value has to be `1`.


Expand All @@ -55,4 +55,4 @@ This is the first release of this `README.md` file.

## Known issues

There are no known issues in this plugin.
There are no known issues in this plugin.

0 comments on commit 8cca9a8

Please sign in to comment.