- Update documentation.
- Add ops estimation for
torch.nn.BatchNorm1dandtorch.nn.BatchNorm2d. - Add
exclude_from_opsoption toModuleProfiler. - Fix
NaNbit sizes and data types for modules without trainable or nontrainable parameters.
torch.nn.BatchNorm1d and torch.nn.BatchNorm2d.exclude_from_ops option to ModuleProfiler.NaN bit sizes and data types for modules without trainable or nontrainable parameters.