the edge_feat is not return of forward method and the edge_attr, which is input of forward, is just returned.
Can I use the edge_feat as the one of the returns of forward method?
Second, Can I predict the h, edge_attr with the EGNN Autoencoder?
In example main_ae.py code, the decoder simply consists of linear transformation and just predicts adjacency matrix.
What I want to do is to predict the h_x (node attribute) and edge attribute. So did I just use the output of autoencoder as same way of adjacency matrix?
Hi,
I wonder about the implementation of two parts in the egnn codes.
First, in E_GCL code,
the edge_feat is not return of
forward
method and the edge_attr, which is input of forward, is just returned. Can I use the edge_feat as the one of the returns offorward
method?Second, Can I predict the
h, edge_attr
with the EGNN Autoencoder? In examplemain_ae.py
code, the decoder simply consists of linear transformation and just predicts adjacency matrix. What I want to do is to predict the h_x (node attribute) and edge attribute. So did I just use the output of autoencoder as same way of adjacency matrix?Best regards, Jinyoung