Hi everyone, I have a question regarding wrapping an operation/module in a CNN model. What I mostly see in codes is people wrap the modules inside custom classes that inherit from torch.nn.Module class. I am curious if it’s necessary to wrap a module in torch.nn.Module class? or it’s only necessary if we want that custom module to be registered and possibly if it requires gradient?