diviswen / PMP-Net

MIT License
46 stars 17 forks source link

issue about main_c3d.py #11

Closed huzijian0622 closed 2 years ago

huzijian0622 commented 2 years ago

Hello, the paper is excellent but there is a problem. When i run 'python main_c3d.py' in PMPplus-Jittor, it shows: Traceback (most recent call last): File "main_c3d.py", line 59, in main() File "main_c3d.py", line 43, in main train_net(cfg) File "/home/lzs/ssd_1t/huzijian/PMP-Net/PMPPlus-Jittor/core/train_c3d.py", line 102, in train_net loss_item = loss.item() RuntimeError: Wrong inputs arguments, Please refer to examples(help(jt.item)).

Types of your inputs are: self = Var, args = (),

The function declarations are: ItemData item()

Failed reason:[f 0530 21:48:53.942659 88 helper_cuda.h:128] CUDA error at /home/lzs/anaconda3/envs/pmp/lib/python3.7/site-packages/jittor/src/mem/allocator.cc:111 code=77( cudaErrorIllegalAddress ) cudaMemcpy(a.ptr, var->mem_ptr, var->size, cudaMemcpyDeviceToHost) [e 0530 21:48:54.275588 88 helper_cuda.h:117] Peek CUDA error at /home/lzs/anaconda3/envs/pmp/lib/python3.7/site-packages/jittor/src/mem/allocator/cuda_dual_allocator.h:101 code=77( cudaErrorIllegalAddress ) _cudaLaunchHostFunc(0, &to_free_allocation, 0) what should i do?