Solved by ptrblck in post #2 
              
              
                You would have to derive your custom Model from nn.Module as: 
class Model(nn.Module):
    def __init__(self, model, pool):
        super().__init__()
        ...

to make sure all nn.Module methods and attributes are available.