meidachen / STPLS3D

🔥 Synthetic and real-world 2d/3d dataset for semantic and instance segmentation (BMVC 2022 Oral)
245 stars 20 forks source link

train question #37

Open WS517 opened 3 months ago

WS517 commented 3 months ago

Thanks for your great work. I tried to train with my own data, the data I dropped into the network is in the format of xyzrgb semantic labels as well as instance labels, but while training the following problem occurs even though I can loop: RuntimeError: cuda runtime error (9) : invalid configuration argument at /pytorch/aten/src/THC/THCTensorMathCompare.cuh:82 How can I solve this problem? 优化后手动分块且训练用网络自己的crop依旧出现问题