[BUG] autogl 0.1.1 is not compactible with torch-geometric 1.6.3 #10
Description
Describe the bug
When I run the graph_classification.py code, I got following error.
How can I fix the error? Any advice will be greatly appreciated.
Thanks
`
~/src/AutoGL/examples$ python graph_classification.py
Traceback (most recent call last):
File "graph_classification.py", line 19, in
cv_split=10,
File "/home/user/src/AutoGL/autogl/solver/classifier/graph_classifier.py", line 294, in fit
self.feature_module.fit(dataset.train_split)
File "/home/user/src/AutoGL/autogl/module/feature/base.py", line 116, in fit
dataset = self._rebuild(dataset, _dataset)
File "/home/user/src/AutoGL/autogl/module/feature/base.py", line 57, in _rebuild
data, slices = dataset.collate(datalist)
File "/home/user/miniconda3/envs/grapl/lib/python3.7/site-packages/torch_geometric/data/in_memory_dataset.py", line 128, in collate
dim=data.cat_dim(key, item))
RuntimeError: Sizes of tensors must match except in dimension 0. Got 1 and 2 in dimension 1 (The offending index is 5)
`
Environment (please complete the following information):
- OS: [Ubuntu20.04]
- python version: [Python 3.7.9]
- autogl version: [0.1.1]