search

BNN-UPC / ignnition

Framework for fast prototyping of Graph Neural Networks
Apache License 2.0
51 stars 16 forks source link

Graphs with 1 edge generate errors #58

Closed HamidL closed 3 years ago

HamidL commented 3 years ago

Training with graphs containing 1 edge generates the following error:

File "[...]/.pyenv/versions/miniconda3-latest/envs/py38/lib/python3.8/site-packages/tensorflow/python/eager/execute.py", line 59, in quick_execute
    tensors = pywrap_tfe.TFE_Py_Execute(ctx._handle, device_name, op_name,
tensorflow.python.framework.errors_impl.InvalidArgumentError:  ConcatOp : Expected concatenating dimensions in the range [-1, 1), but got 1
     [[{{node gnn_model/StatefulPartitionedCall/ignnition_model/message_passing/iteration_0/stage_0/MP_to_AS/message_phase/AS_to_AS/create_message_AS_to_AS/apply_nn_0/concat_2}}]] [Op:__inference_train_function_8025]

TensorFlow version is 2.5.0 and Python version is 3.8.5

MiquelFerriol commented 3 years ago

This issue has been solved in PR #78