Dear authors,
I added to the library argument options to be used in the evaluation of any module based on the graph blocks. These arguments aim to make possible the use of layers like Dropout or Batch Normalization inside the graph blocks to process either edge, node or global features. I also implemented some tests to validate my implementation, and I change the documentation as needed.
I hope this simple feature could be usefull to improve graph-nets framework.
Dear authors, I added to the library argument options to be used in the evaluation of any module based on the graph blocks. These arguments aim to make possible the use of layers like Dropout or Batch Normalization inside the graph blocks to process either edge, node or global features. I also implemented some tests to validate my implementation, and I change the documentation as needed.
I hope this simple feature could be usefull to improve graph-nets framework.
Best, Caio.