gru = GRU(input_size=1, hidden_size=256, output_size=1)