def __init__(self):
    super().__init__()
    self.reset_seed()
    self.elem = math.prod(Encoder().output_size)
    self.input_size = (1, self.elem)
    self.output_size = (1, 1)
    self.dense1 = Linear(self.elem, 512)
    self.norm1 = LayerNorm(512)
    self.dense2 = Linear(512, 256)
    self.norm2 = LayerNorm(256)
    self.dense3 = Linear(256, 1)
def __init__(self):
    super().__init__()
    self.reset_seed()
    self.elem = math.prod(Encoder().output_size)
    self.input_size = (1, self.elem)
    self.output_size = (1, 1)
    self.dense1 = Linear(self.elem, 512)
    self.norm1 = LayerNorm(512)
    self.dense2 = Linear(512, 256)
    self.norm2 = LayerNorm(256)
    self.dense3 = Linear(256, 1)