我注释掉了loader.py中的

def download(self):

#     raise NotImplementedError('Must indicate valid location of raw data. '
#                               'No download allowed')

再次运行pretrain_GEOM
Traceback (most recent call last):
File "pretraining.py", line 168, in
main()
File "pretraining.py", line 157, in main
train_mae(args, model, loader, optimizer, epoch)
File "pretraining.py", line 26, in train_mae
loss = model(batch)
File "/home/penghuan/miniconda3/envs/calm/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1110, in _call_impl
return forward_call(*input, **kwargs)
File "/home/penghuan/code/SimSGT/regression/model.py", line 862, in forward
h = self.encoder(self.gnn_act(h), edge_index, edge_attr, data.batch, data.mask_tokens, pe_tokens)
File "/home/penghuan/miniconda3/envs/calm/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1110, in _call_impl
return forward_call(*input, **kwargs)
File "/home/penghuan/code/SimSGT/regression/model.py", line 1559, in forward
h = self.gnns[layer](h_list[layer], edge_index, edge_attr)
File "/home/penghuan/miniconda3/envs/calm/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1110, in _call_impl
return forward_call(*input, **kwargs)
File "/home/penghuan/code/SimSGT/regression/model.py", line 291, in forward
return self.propagate(edge_index, x=x, edge_attr=edge_embeddings)
File "/home/penghuan/miniconda3/envs/calm/lib/python3.8/site-packages/torch_geometric/nn/conv/message_passing.py", line 351, in propagate
out = self.update(out, **update_kwargs)
File "/home/penghuan/code/SimSGT/regression/model.py", line 297, in update
return self.mlp(aggr_out)
File "/home/penghuan/miniconda3/envs/calm/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1110, in _call_impl
return forward_call(*input, **kwargs)
File "/home/penghuan/miniconda3/envs/calm/lib/python3.8/site-packages/torch/nn/modules/container.py", line 141, in forward
input = module(input)
File "/home/penghuan/miniconda3/envs/calm/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1110, in _call_impl
return forward_call(*input, **kwargs)
File "/home/penghuan/miniconda3/envs/calm/lib/python3.8/site-packages/torch/nn/modules/activation.py", line 98, in forward
return F.relu(input, inplace=self.inplace)
File "/home/penghuan/miniconda3/envs/calm/lib/python3.8/site-packages/torch/nn/functional.py", line 1442, in relu
result = torch.relu(input)
RuntimeError: CUDA error: device-side assert triggered
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.

from simsgt.

Traceback (most recent call last):
File "pretraining.py", line 168, in
main()
File "pretraining.py", line 157, in main
train_mae(args, model, loader, optimizer, epoch)
File "pretraining.py", line 26, in train_mae
loss = model(batch)
File "/home/penghuan/miniconda3/envs/calm/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1110, in _call_impl
return forward_call(*input, **kwargs)
File "/home/penghuan/SimSGT/regression/model.py", line 857, in forward
h = self.encoder(self.gnn_act(h), edge_index, edge_attr, data.batch, data.mask_tokens, pe_tokens)
File "/home/penghuan/miniconda3/envs/calm/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1110, in _call_impl
return forward_call(*input, **kwargs)
File "/home/penghuan/SimSGT/regression/model.py", line 1554, in forward
h = self.gnns[layer](h_list[layer], edge_index, edge_attr)
File "/home/penghuan/miniconda3/envs/calm/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1110, in _call_impl
return forward_call(*input, **kwargs)
File "/home/penghuan/SimSGT/regression/model.py", line 291, in forward
return self.propagate(edge_index, x=x, edge_attr=edge_embeddings)
File "/home/penghuan/miniconda3/envs/calm/lib/python3.8/site-packages/torch_geometric/nn/conv/message_passing.py", line 317, in propagate
out = self.message(**msg_kwargs)
File "/home/penghuan/SimSGT/regression/model.py", line 294, in message
return self.activation(x_j + edge_attr)
File "/home/penghuan/miniconda3/envs/calm/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1110, in _call_impl
return forward_call(*input, **kwargs)
File "/home/penghuan/miniconda3/envs/calm/lib/python3.8/site-packages/torch/nn/modules/activation.py", line 1111, in forward
return F.prelu(input, self.weight)
RuntimeError: CUDA error: device-side assert triggered
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.

from simsgt.

每一次的报错内容还不一定一样 希望能解答一下 是数据格式的问题或者其他方面的问题吗？

from simsgt.