Open ziweiyang opened 1 year ago
This sounds like one of the vectors of features you are building (e.g. node features or edge features), have zero dimensionality. So this is trying to instantiate a weight of size [0, N], and it is failing when running the initialization for that weight.
Hi @alvarosg , thanks for your comment. I figured it out like what you mentioned here. Thanks for sharing this library, which saves my lots of work.
Best, ZY
Hi developer,
Thanks for sharing this work. I am trying to reproduce the MeshGraphNet developed by Deepminds in 2021, by using the module.EncodeProcessDecode and module.InteractionNetwroks.
I am using demo.tf2 as prototype and replace self._core with module.InteractionNetwork. I am only using test.tfrecord in original cylinder_flow case for easy testing the model on colab. (There is no global features, and num_nodes/num_edges equal to num_vertices/num_edges in that moment of mesh-space). I define each individual timestep as a
GraphTuple
. Everything works fine until I started the actual training loop, then I got this error message:StagingError: in user code:
I am not sure where I should look at to solve this. Any ideas?
Best, ZY