can't seem to run "python train.py"

Question

can't seem to run "python train.py"

furycurry opened this issue 3 years ago · 4 comments

When I am trying to run python train.py in the terminal, there is an error telling
mat1 and mat2 shapes cannot be multiplied (64x1024 and 64x2).
I am using Jupyter Notebook and have successfully downloaded and installed all the prerequisites for the code to run.

Here is also a screenshot of the error.

@YingtongDou

Answer 1 · 2022-01-29T22:03:33.000Z

The shape of the first matrix should be 1024 x 64 (batch_size x emb_size), please check your code and make sure the GNN output has the above shape.

Answer 2 · 2022-02-06T12:24:52.000Z

same problem. just modified the shape of mat1 using torch.t, but the training results had little classification performance. looks like the code needs to be fixed.

Answer 3 · 2022-02-06T23:13:09.000Z

Sorry for the inconvenience. The problems are fixed in the last update #9. If there are any further questions, please don't hesitate to contact us.

Answer 4 · 2022-02-07T01:32:57.000Z

same problem. just modified the shape of mat1 using torch.t, but the training results had little classification performance. looks like the code needs to be fixed.

Thanks