question about the data format

[llllly26]

Hi @lucidrains,

I found that the input of the En-Transformer is different from EGNN in your released repository that EGNN using torch_geometric lib to deal with the node and edge. For instance, EGNN-pytorch encode node: [bs_node_number_sum, dim], and edge: [2, bs_edge_number_sum]. So I want to know how to put this type data into En-Transformer?

thanks!