Update type hint for `register_notrace_module`
Update type hint for `register_notrace_module`

register_notrace_module is used to decorate types (i.e. subclasses of nn.Module).
It is not called on module instances.
rwightman/pytorch-image-models,rwightman/pytorch-image-models