Hi,
I am new in CNN, also PyTorch. I am trying to run this project:
https://github.com/theswgong/MoNet/tree/master/manifolds 2
My machine configuration is: core i7, 8GB RAM. And graphics configuration is:
nvidia-smi789×498 55.9 KB
The error I am getting is:

Data(edge_attr=[41328, 3], edge_index=[2, 41328], pos=[6890, 3], x=[6890, 3], y=[1])
MoNet(
(fc0): Linear(in_features=3, out_features=16, bias=True)
  (conv1): GMMConv(16, 32, kernel_size=10)
  (conv2): GMMConv(32, 64, kernel_size=10)
  (conv3): GMMConv(64, 128, kernel_size=10)
  (fc1): Linear(in_features=128, out_features=256, bias=True)
  (fc2): Linear(in_features=256, out_features=6890, bias=True)
)
Traceback (most recent call last):
  File "/home/nuha/anaconda3/envs/monet/lib/python3.7/runpy.py", line 193, in _run_module_as_main
    "__main__", mod_spec)
  File "/home/nuha/anaconda3/envs/monet/lib/python3.7/runpy.py", line 85, in _run_code
    exec(code, run_globals)
  File "/home/nuha/Documents/MoNet/manifolds/main.py", line 110, in <module>
    optimizer, scheduler, device)
  File "/home/nuha/Documents/MoNet/manifolds/train_eval.py", line 19, in run
    train_loss = train(model, train_loader, target, optimizer, device)
  File "/home/nuha/Documents/MoNet/manifolds/train_eval.py", line 41, in train
    loss = F.nll_loss(model(data.to(device)), target)
  File "/home/nuha/anaconda3/envs/monet/lib/python3.7/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
    result = self.forward(*input, **kwargs)
  File "/home/nuha/Documents/MoNet/manifolds/main.py", line 91, in forward
    x = F.elu(self.conv1(x, edge_index, edge_attr))
  File "/home/nuha/anaconda3/envs/monet/lib/python3.7/site-packages/torch/nn/modules/module.py", line 727, in _call_impl
    result = self.forward(*input, **kwargs)
  File "/home/nuha/Documents/MoNet/conv/gmm_conv.py", line 47, in forward
    out = self.propagate(edge_index, x=out, pseudo=pseudo)
  File "/home/nuha/anaconda3/envs/monet/lib/python3.7/site-packages/torch_geometric/nn/conv/message_passing.py", line 234, in propagate
    kwargs)
  File "/home/nuha/anaconda3/envs/monet/lib/python3.7/site-packages/torch_geometric/nn/conv/message_passing.py", line 158, in __collect__
    j if arg[-2:] == '_j' else i)
  File "/home/nuha/anaconda3/envs/monet/lib/python3.7/site-packages/torch_geometric/nn/conv/message_passing.py", line 127, in __lift__
    return src.index_select(self.node_dim, index)
RuntimeError: CUDA out of memory. Tried to allocate 33.95 GiB (GPU 0; 3.95 GiB total capacity; 17.50 MiB already allocated; 3.13 GiB free; 24.00 MiB reserved in total by PyTorch)
[1]    308878 segmentation fault (core dumped)  python -m manifolds.main

`
I tried reducing the batch size. But That introduced new problem as:

The size of tensor a (22065) must match the size of tensor b (25) at non-singleton dimension 1