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.