def call(self, x, training=False):
    h = self.act(self.c_fc(x))
    h2 = self.c_proj(h)
    h2 = self.dropout(h2, training=training)
    return h2
def call(self, x, training=False):
    h = self.act(self.c_fc(x))
    h2 = self.c_proj(h)
    h2 = self.dropout(h2, training=training)
    return h2