lightaime / deep_gcns_torch

Pytorch Repo for DeepGCNs (ICCV'2019 Oral, TPAMI'2021), DeeperGCN (arXiv'2020) and GNN1000(ICML'2021): https://www.deepgcns.org
MIT License
1.13k stars 155 forks source link

error occurs in training modelnet40 after epoch 20 #74

Closed amiltonwong closed 3 years ago

amiltonwong commented 3 years ago

Hi, @lightaime ,

I got the error RuntimeError: CUDA error: device-side assert triggered, after training modelnet40 in epoch 20 (details in log.txt)

Could you give me some hints on how to fix this issue?

Thanks~

guochengqian commented 3 years ago

hi @amiltonwong no worry, I will check it as soon as possible.

guochengqian commented 3 years ago

hi @amiltonwong. Please use the most recent code. I make the batch_index_select more stable. I train them just fine and easily reproduce the results.