Cisco-Talos / binary_function_similarity

MIT License
262 stars 31 forks source link

The Problem in the GGSNN-GMN Model #40

Open liaoluoxing opened 9 months ago

liaoluoxing commented 9 months ago

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.