Method __init__

utils/spatial_att_github.py:184–189

(self)

class QueryNetwork(nn.Module):
    def __init__(self):
        super(QueryNetwork, self).__init__()
        # TODO: Add proper non-linearity.
        self.model = nn.Sequential(
            nn.Linear(256, 128), nn.ReLU(), nn.Linear(128, 288), nn.ReLU(), nn.Linear(288, 288)
        )

    def forward(self, query):
        out = self.model(query)