The Problem in the GGSNN-GMN Model

[liaoluoxing]

Hello, dear author.
I meet some problem in the Part 2 of GGSNN-GMN(binary_function_similarity/Models
/GGSNN-GMN/).

When i run the command "Example: run the validation on Dataset-1 using the model_checkpoint that we trained on Dataset-1", there are some errors. The following code is the error message:

Traceback (most recent call last):
  File "/code/gnn.py", line 219, in <module>
    main()
  File "/code/gnn.py", line 209, in main
    model_validate(config)
  File "/code/gnn.py", line 110, in model_validate
    gnn_model.model_validate()
  File "/code/core/gnn_model.py", line 289, in model_validate
    self._model_initialize(training_set, is_training=False)
  File "/code/core/gnn_model.py", line 144, in _model_initialize
    self._create_network(batch_generator, is_training=is_training)
  File "/code/core/gnn_model.py", line 119, in _create_network
    node_feature_dim = first_batch_graphs.node_features.shape[-1]
AttributeError: 'tuple' object has no attribute 'node_features'

I suspect that I did some things wrongly. If you can help me, I would be very appreciate. Thank you a lot.