RuntimeError: mat1 and mat2 shapes cannot be multiplied (880x598 and 714x128)

[LinguaLogician]

Epoch 1: 0%| | 0/1251 [00:00<?, ?it/s]
Traceback (most recent call last):
File "/home/liangtao.wu/Softwares/miniconda3/envs/retroxpert/lib/python3.10/runpy.py", line 196, in _run_module_as_main
return _run_code(code, main_globals, None,
File "/home/liangtao.wu/Softwares/miniconda3/envs/retroxpert/lib/python3.10/runpy.py", line 86, in _run_code
exec(code, run_globals)
File "/home/liangtao.wu/.vscode-server/extensions/ms-python.debugpy-2024.12.0-linux-x64/bundled/libs/debugpy/adapter/../../debugpy/launcher/../../debugpy/main.py", line 71, in
cli.main()
File "/home/liangtao.wu/.vscode-server/extensions/ms-python.debugpy-2024.12.0-linux-x64/bundled/libs/debugpy/adapter/../../debugpy/launcher/../../debugpy/../debugpy/server/cli.py", line 501, in main
run()
File "/home/liangtao.wu/.vscode-server/extensions/ms-python.debugpy-2024.12.0-linux-x64/bundled/libs/debugpy/adapter/../../debugpy/launcher/../../debugpy/../debugpy/server/cli.py", line 351, in run_file
runpy.run_path(target, run_name="main")
File "/home/liangtao.wu/.vscode-server/extensions/ms-python.debugpy-2024.12.0-linux-x64/bundled/libs/debugpy/_vendored/pydevd/_pydevd_bundle/pydevd_runpy.py", line 310, in run_path
return _run_module_code(code, init_globals, run_name, pkg_name=pkg_name, script_name=fname)
File "/home/liangtao.wu/.vscode-server/extensions/ms-python.debugpy-2024.12.0-linux-x64/bundled/libs/debugpy/_vendored/pydevd/_pydevd_bundle/pydevd_runpy.py", line 127, in _run_module_code
_run_code(code, mod_globals, init_globals, mod_name, mod_spec, pkg_name, script_name)
File "/home/liangtao.wu/.vscode-server/extensions/ms-python.debugpy-2024.12.0-linux-x64/bundled/libs/debugpy/_vendored/pydevd/_pydevd_bundle/pydevd_runpy.py", line 118, in _run_code
exec(code, run_globals)
File "/data/liangtao/DemoSpace/RetroXpert/train.py", line 334, in
h_pred, e_pred = GAT_model(g_dgl, x_atom)
File "/home/liangtao.wu/Softwares/miniconda3/envs/retroxpert/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1553, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/liangtao.wu/Softwares/miniconda3/envs/retroxpert/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1562, in _call_impl
return forward_call(*args, **kwargs)
File "/data/liangtao/DemoSpace/RetroXpert/model/gat.py", line 121, in forward
h, _ = self.gat[l](g, h, merge='flatten')
File "/home/liangtao.wu/Softwares/miniconda3/envs/retroxpert/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1553, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/liangtao.wu/Softwares/miniconda3/envs/retroxpert/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1562, in _call_impl
return forward_call(*args, **kwargs)
File "/data/liangtao/DemoSpace/RetroXpert/model/gat.py", line 72, in forward
outs = list(map(lambda x: x(g, h), self.heads))
File "/data/liangtao/DemoSpace/RetroXpert/model/gat.py", line 72, in
outs = list(map(lambda x: x(g, h), self.heads))
File "/home/liangtao.wu/Softwares/miniconda3/envs/retroxpert/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1553, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/liangtao.wu/Softwares/miniconda3/envs/retroxpert/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1562, in _call_impl
return forward_call(*args, **kwargs)
File "/data/liangtao/DemoSpace/RetroXpert/model/gat.py", line 45, in forward
g.ndata['h'] = self.embed_node(h)
File "/home/liangtao.wu/Softwares/miniconda3/envs/retroxpert/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1553, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/liangtao.wu/Softwares/miniconda3/envs/retroxpert/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1562, in _call_impl
return forward_call(*args, **kwargs)
File "/home/liangtao.wu/Softwares/miniconda3/envs/retroxpert/lib/python3.10/site-packages/torch/nn/modules/linear.py", line 117, in forward
return F.linear(input, self.weight, self.bias)
RuntimeError: mat1 and mat2 shapes cannot be multiplied (797x598 and 714x128)