Hi everyone,
I am wondering whether it is ok to define additional function in an nn.Module:
class MyModule(nn.Module):
    ...
    def classify(...):
        ...
    def detect(...):
        ....
    def forward(...):
        ....

My question is based on this previous thread: Any different between model(input) and model.forward(input) which says that one should not call forward directly because some hooks are not used then. So is it discouraged define additional methods? Putting everything in forward() with some flag would be an alternative, but an ugly one.