Hi,it seems that the relation encoder GRU cannot support batch_size, I just modify the code of the relation encoder is enough? and why you didn't use batch_size such as 32 in training for acceleration