lbadams2

class GRU(nn.Module):
    INPUT_SIZE = 603
    HIDDEN_SIZE = 100
    OUTPUT_SIZE = 1

    def __init__(self):
        super(GRU, self).__init__()
        self.gru = nn.GRU(self.INPUT_SIZE, self.HIDDEN_SIZE)
        self.linear = nn.Linear(self.HIDDEN_SIZE, self.OUTPUT_SIZE)
        self.sm = nn.Sigmoid()